ArmNN
NotReleased
|
#include <RefElementwiseWorkload.hpp>
Public Types | |
using | InType = typename ElementwiseBinaryFunction< Functor >::InType |
using | OutType = typename ElementwiseBinaryFunction< Functor >::OutType |
Public Member Functions | |
RefElementwiseWorkload (const ParentDescriptor &descriptor, const WorkloadInfo &info) | |
void | PostAllocationConfigure () override |
void | Execute () const override |
Public Member Functions inherited from BaseWorkload< ParentDescriptor > | |
BaseWorkload (const ParentDescriptor &descriptor, const WorkloadInfo &info) | |
void | PostAllocationConfigure () override |
const ParentDescriptor & | GetData () const |
profiling::ProfilingGuid | GetGuid () const final |
Public Member Functions inherited from IWorkload | |
virtual | ~IWorkload () |
virtual void | RegisterDebugCallback (const DebugCallbackFunction &) |
Additional Inherited Members | |
Protected Attributes inherited from BaseWorkload< ParentDescriptor > | |
const ParentDescriptor | m_Data |
const profiling::ProfilingGuid | m_Guid |
Definition at line 21 of file RefElementwiseWorkload.hpp.
using InType = typename ElementwiseBinaryFunction<Functor>::InType |
Definition at line 24 of file RefElementwiseWorkload.hpp.
using OutType = typename ElementwiseBinaryFunction<Functor>::OutType |
Definition at line 25 of file RefElementwiseWorkload.hpp.
RefElementwiseWorkload | ( | const ParentDescriptor & | descriptor, |
const WorkloadInfo & | info | ||
) |
Definition at line 21 of file RefElementwiseWorkload.cpp.
|
overridevirtual |
Implements IWorkload.
Definition at line 41 of file RefElementwiseWorkload.cpp.
References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, TensorInfo::GetShape(), armnn::GetTensorInfo(), StringMapping::Instance(), BaseWorkload< ParentDescriptor >::m_Data, StringMapping::RefAdditionWorkload_Execute, StringMapping::RefDivisionWorkload_Execute, StringMapping::RefMaximumWorkload_Execute, StringMapping::RefMinimumWorkload_Execute, StringMapping::RefMultiplicationWorkload_Execute, and StringMapping::RefSubtractionWorkload_Execute.
|
overridevirtual |
Implements IWorkload.
Definition at line 29 of file RefElementwiseWorkload.cpp.
References armnn::GetTensorInfo(), and BaseWorkload< ParentDescriptor >::m_Data.