ArmNN
 22.02
Public Types | Public Member Functions | List of all members
RefElementwiseWorkload< Functor, ParentDescriptor, DebugString > Class Template Reference

#include <RefElementwiseWorkload.hpp>

Inheritance diagram for RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >:
RefBaseWorkload< ParentDescriptor > BaseWorkload< ParentDescriptor > IWorkload

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
 
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 >
ParentDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 

Detailed Description

template<typename Functor, typename ParentDescriptor, typename armnn::StringMapping::Id DebugString>
class armnn::RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >

Definition at line 21 of file RefElementwiseWorkload.hpp.

Member Typedef Documentation

◆ InType

using InType = typename ElementwiseBinaryFunction<Functor>::InType

Definition at line 24 of file RefElementwiseWorkload.hpp.

◆ OutType

using OutType = typename ElementwiseBinaryFunction<Functor>::OutType

Definition at line 25 of file RefElementwiseWorkload.hpp.

Constructor & Destructor Documentation

◆ RefElementwiseWorkload()

RefElementwiseWorkload ( const ParentDescriptor &  descriptor,
const WorkloadInfo &  info 
)

Definition at line 21 of file RefElementwiseWorkload.cpp.

24  : RefBaseWorkload<ParentDescriptor>(desc, info)
25 {
26 }

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 29 of file RefElementwiseWorkload.cpp.

References BaseWorkload< ParentDescriptor >::m_Data.

Referenced by RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >::ExecuteAsync().

30 {
31  Execute(m_Data.m_Inputs, m_Data.m_Outputs);
32 }

◆ ExecuteAsync()

void ExecuteAsync ( WorkingMemDescriptor &  workingMemDescriptor)
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.

37 {
38  Execute(workingMemDescriptor.m_Inputs, workingMemDescriptor.m_Outputs);
39 }
The documentation for this class was generated from the following files: