ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Reduce.h File Reference
#include "cker/Shape.h"
#include "cker/Types.h"
#include "cker/Utils.h"
#include "cker/neon/neon_check.h"

Go to the source code of this file.

Data Structures

class  nnfw::cker::Reduce
 

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

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)