Select.h File Reference

#include "cker/Shape.h"
#include "cker/Utils.h"
#include <cmath>

Go to the source code of this file.

Namespaces
namespace	nnfw
namespace	nnfw::cker

Functions
template<typename D , typename T >
void	nnfw::cker::Select (const Shape &input_condition_shape, const D *input_condition_data, const Shape &input_x_shape, const T *input_x_data, const Shape &input_y_shape, const T *input_y_data, const Shape &output_shape, T *output_data)
template<typename D , typename T >
void	nnfw::cker::RankOneSelect (const Shape &input_condition_shape, const D *input_condition_data, const Shape &input_x_shape, const T *input_x_data, const Shape &input_y_shape, const T *input_y_data, const Shape &output_shape, T *output_data)
template<typename D , typename T >
void	nnfw::cker::BroadcastSelect4DSlow (const Shape &input_condition_shape, const D *input_condition_data, const Shape &input_x_shape, const T *input_x_data, const Shape &input_y_shape, const T *input_y_data, const Shape &output_shape, T *output_data)