Go to the source code of this file.
|
template<typename In , typename Out > |
bool | nnfw::cker::ReduceImpl (const In *input_data, const Shape &input_shape, const Shape &, const int *axis, const int num_axis, int *input_iter, Out reducer(const Out current, const In in), Out *output_data) |
|
bool | nnfw::cker::ResolveAxis (const int num_dims, const std::vector< int > &axes, int *out_axis, int *out_num_axis) |
|
template<typename T > |
bool | nnfw::cker::InitTensorDataForReduce (const Shape &shape, const T init_value, T *data) |
|