ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Static Public Member Functions
nnfw::cker::functor::ReduceFunctor< Device, Reducer > Struct Template Reference

#include <Einsum.h>

Static Public Member Functions

template<typename OUT_T , typename IN_T , typename ReductionAxes >
static void Reduce (const Device &d, OUT_T out, IN_T in, const ReductionAxes &reduction_axes, const Reducer &reducer)
 

Detailed Description

template<typename Device, typename Reducer>
struct nnfw::cker::functor::ReduceFunctor< Device, Reducer >

Definition at line 85 of file Einsum.h.

Member Function Documentation

◆ Reduce()

template<typename Device , typename Reducer >
template<typename OUT_T , typename IN_T , typename ReductionAxes >
static void nnfw::cker::functor::ReduceFunctor< Device, Reducer >::Reduce ( const Device &  d,
OUT_T  out,
IN_T  in,
const ReductionAxes &  reduction_axes,
const Reducer &  reducer 
)
inlinestatic

Definition at line 88 of file Einsum.h.

90 {
91 out.device(d) = in.reduce(reduction_axes, reducer);
92 }
The documentation for this struct was generated from the following file: