ArmNN
 24.02
ElementwiseUnaryFunction< Functor > Struct Template Reference

#include <ElementwiseFunction.hpp>

Public Types

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

Public Member Functions

 ElementwiseUnaryFunction (const TensorShape &inShape, const TensorShape &outShape, Decoder< InType > &inData, Encoder< OutType > &outData)
 

Detailed Description

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

Definition at line 29 of file ElementwiseFunction.hpp.

Member Typedef Documentation

◆ InType

using InType = typename Functor::argument_type

Definition at line 32 of file ElementwiseFunction.hpp.

◆ OutType

using OutType = typename Functor::result_type

Definition at line 31 of file ElementwiseFunction.hpp.

Constructor & Destructor Documentation

◆ ElementwiseUnaryFunction()

ElementwiseUnaryFunction ( const TensorShape inShape,
const TensorShape outShape,
Decoder< InType > &  inData,
Encoder< OutType > &  outData 
)

Definition at line 36 of file ElementwiseFunction.cpp.

40 {
41  BroadcastLoop(inShape, outShape).Unroll(Functor(), 0, inData, outData);
42 }

References BroadcastLoop::Unroll().


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