ArmNN
 24.02
ElementwiseBinaryFunction< Functor > Struct Template Reference

#include <ElementwiseFunction.hpp>

Public Types

using OutType = typename Functor::result_type
 
using InType = typename Functor::first_argument_type
 

Public Member Functions

 ElementwiseBinaryFunction (const TensorShape &inShape0, const TensorShape &inShape1, const TensorShape &outShape, Decoder< InType > &inData0, Decoder< InType > &inData1, Encoder< OutType > &outData)
 

Detailed Description

template<typename Functor>
struct armnn::ElementwiseBinaryFunction< Functor >

Definition at line 15 of file ElementwiseFunction.hpp.

Member Typedef Documentation

◆ InType

using InType = typename Functor::first_argument_type

Definition at line 18 of file ElementwiseFunction.hpp.

◆ OutType

using OutType = typename Functor::result_type

Definition at line 17 of file ElementwiseFunction.hpp.

Constructor & Destructor Documentation

◆ ElementwiseBinaryFunction()

ElementwiseBinaryFunction ( const TensorShape inShape0,
const TensorShape inShape1,
const TensorShape outShape,
Decoder< InType > &  inData0,
Decoder< InType > &  inData1,
Encoder< OutType > &  outData 
)

Definition at line 25 of file ElementwiseFunction.cpp.

31 {
32  BroadcastLoop(inShape0, inShape1, outShape).Unroll(Functor(), 0, inData0, inData1, outData);
33 }

References BroadcastLoop::Unroll().


The documentation for this struct was generated from the following files: