ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert_micro::execute::pal::Reducer< T, ReduceFn > Class Template Reference

#include <PALReduceCommon.h>

Public Member Functions

 Reducer (core::OMReduceDataContext< T > &ctx, T init_value)
 
bool Mean ()
 
bool Reduce ()
 

Detailed Description

template<typename T, template< typename > class ReduceFn>
class onert_micro::execute::pal::Reducer< T, ReduceFn >

Definition at line 76 of file PALReduceCommon.h.

Constructor & Destructor Documentation

◆ Reducer()

template<typename T , template< typename > class ReduceFn>
onert_micro::execute::pal::Reducer< T, ReduceFn >::Reducer ( core::OMReduceDataContext< T > &  ctx,
init_value 
)
inlineexplicit

Definition at line 93 of file PALReduceCommon.h.

94 : _input(ctx.Input())
95 , _output(ctx.Output())
96 , _axes(ctx.Axis())
97 , _init_value(init_value)
98 {}

Member Function Documentation

◆ Mean()

template<typename T , template< typename > class ReduceFn>
bool onert_micro::execute::pal::Reducer< T, ReduceFn >::Mean ( )
inline

Definition at line 101 of file PALReduceCommon.h.

102 {
103 if (SpecialCaseMeanImpl())
104 return true;
105
106 return ReduceImpl(true);
107 }

Referenced by impl::CircleMean().

◆ Reduce()

template<typename T , template< typename > class ReduceFn>
bool onert_micro::execute::pal::Reducer< T, ReduceFn >::Reduce ( )
inline

Definition at line 109 of file PALReduceCommon.h.

110 {
111 return ReduceImpl();
112 }

Referenced by impl::CircleReduceMax(), impl::CircleReduceProd(), and impl::CircleSum().


The documentation for this class was generated from the following file: