22.05
|
#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 | Execute () const override |
void | ExecuteAsync (WorkingMemDescriptor &workingMemDescriptor) override |
Public Member Functions inherited from RefBaseWorkload< ParentDescriptor > | |
RefBaseWorkload (const ParentDescriptor &descriptor, const WorkloadInfo &info) | |
virtual bool | SupportsTensorHandleReplacement () const override |
void | ReplaceInputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override |
void | ReplaceOutputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override |
Public Member Functions inherited from BaseWorkload< ParentDescriptor > | |
BaseWorkload (const ParentDescriptor &descriptor, const WorkloadInfo &info) | |
void | ExecuteAsync (WorkingMemDescriptor &workingMemDescriptor) override |
void | PostAllocationConfigure () override |
const ParentDescriptor & | GetData () const |
arm::pipe::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 > | |
ParentDescriptor | m_Data |
const arm::pipe::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 29 of file RefElementwiseWorkload.cpp.
References BaseWorkload< ParentDescriptor >::m_Data.
Referenced by RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >::ExecuteAsync().
|
overridevirtual |
Implements IWorkload.
Definition at line 35 of file RefElementwiseWorkload.cpp.
References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >::Execute(), TensorInfo::GetShape(), armnn::GetTensorInfo(), StringMapping::Instance(), WorkingMemDescriptor::m_Inputs, WorkingMemDescriptor::m_Outputs, StringMapping::RefAdditionWorkload_Execute, StringMapping::RefDivisionWorkload_Execute, StringMapping::RefMaximumWorkload_Execute, StringMapping::RefMinimumWorkload_Execute, StringMapping::RefMultiplicationWorkload_Execute, and StringMapping::RefSubtractionWorkload_Execute.