ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALReduceCommon.h File Reference
#include "PALUtils.h"
#include "core/OMCustomRuntimeData.h"
#include "core/OMRuntimeShape.h"

Go to the source code of this file.

Data Structures

struct  onert_micro::execute::pal::ReduceSumFn< T >
 
struct  onert_micro::execute::pal::ReduceSumFn< int8_t >
 
struct  onert_micro::execute::pal::ReduceProductFn< T >
 
struct  onert_micro::execute::pal::ReduceProductFn< int8_t >
 

Namespaces

namespace  onert_micro
 
namespace  onert_micro::execute
 
namespace  onert_micro::execute::pal
 

Functions

bool onert_micro::execute::pal::resolveAxis (const int num_dims, const int *axis, const int64_t num_axis, int *out_axis, int *out_num_axis)
 
template<typename T >
bool onert_micro::execute::pal::ReduceGeneric (const T *input_data, const int *input_dims, const int input_num_dims, T *output_data, const int *axis, const int64_t num_axis_dimensions, T init_value, const int output_flat_size, T reducer(const T, const T))
 
template<typename T >
bool onert_micro::execute::pal::reduceSumImpl (const T *input_data, const int *input_dims, const int input_num_dims, T *output_data, const int *axis, const int num_axis, const int num_outputs)
 
template<typename T >
void onert_micro::execute::pal::MeanROWH (const OMRuntimeShape &unextended_input_shape, const T *input_data, const OMRuntimeShape &unextended_output_shape, T *output_data)
 
template<typename T , template< typename > class ReduceFn>
bool onert_micro::execute::pal::ReduceGeneric (OMReduceDataContext< T > &ctx)
 
template<typename T , template< typename > class ReduceFn>
bool onert_micro::execute::pal::Reduce (OMReduceDataContext< T > &ctx, bool mean=false)