ArmNN
 22.08
RefDebugWorkload< DataType > Class Template Reference

#include <RefDebugWorkload.hpp>

Inheritance diagram for RefDebugWorkload< DataType >:
TypedWorkload< DebugQueueDescriptor, DataType > BaseWorkload< DebugQueueDescriptor > IWorkload

Public Member Functions

 RefDebugWorkload (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void Execute () const override
 
void ExecuteAsync (ExecutionData &executionData) override
 
void RegisterDebugCallback (const DebugCallbackFunction &func) override
 
- Public Member Functions inherited from TypedWorkload< DebugQueueDescriptor, DataType >
 TypedWorkload (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info)
 
- Public Member Functions inherited from BaseWorkload< DebugQueueDescriptor >
 BaseWorkload (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void ExecuteAsync (ExecutionData &executionData) override
 
void PostAllocationConfigure () override
 
const DebugQueueDescriptorGetData () const
 
arm::pipe::ProfilingGuid GetGuid () const final
 
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 IWorkload
virtual ~IWorkload ()
 
virtual armnn::Optional< armnn::MemoryRequirementsGetMemoryRequirements ()
 

Static Public Member Functions

static const std::string & GetName ()
 

Additional Inherited Members

- Protected Attributes inherited from BaseWorkload< DebugQueueDescriptor >
DebugQueueDescriptor m_Data
 
const arm::pipe::ProfilingGuid m_Guid
 

Detailed Description

template<armnn::DataType DataType>
class armnn::RefDebugWorkload< DataType >

Definition at line 16 of file RefDebugWorkload.hpp.

Constructor & Destructor Documentation

◆ RefDebugWorkload()

RefDebugWorkload ( const DebugQueueDescriptor descriptor,
const WorkloadInfo info 
)
inline

Definition at line 19 of file RefDebugWorkload.hpp.

20  : TypedWorkload<DebugQueueDescriptor, DataType>(descriptor, info)
21  , m_Callback(nullptr) {}

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 18 of file RefDebugWorkload.cpp.

Referenced by RefDebugWorkload< DataType >::ExecuteAsync(), and RefDebugWorkload< DataType >::GetName().

19 {
21 }
void Execute() const override
std::vector< ITensorHandle * > m_Inputs

◆ ExecuteAsync()

void ExecuteAsync ( ExecutionData executionData)
overridevirtual

Implements IWorkload.

Definition at line 24 of file RefDebugWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, armnn::Debug(), RefDebugWorkload< DataType >::Execute(), TensorInfo::GetNumElements(), armnn::GetTensorInfo(), ExecutionData::m_Data, and WorkingMemDescriptor::m_Inputs.

Referenced by RefDebugWorkload< DataType >::GetName().

25 {
26  WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
27  Execute(workingMemDescriptor->m_Inputs);
28 }
void Execute() const override

◆ GetName()

static const std::string& GetName ( )
inlinestatic

Definition at line 23 of file RefDebugWorkload.hpp.

References RefDebugWorkload< DataType >::Execute(), RefDebugWorkload< DataType >::ExecuteAsync(), armnn::GetDataTypeName(), and RefDebugWorkload< DataType >::RegisterDebugCallback().

24  {
25  static const std::string name = std::string("RefDebug") + GetDataTypeName(DataType) + "Workload";
26  return name;
27  }
constexpr const char * GetDataTypeName(DataType dataType)
Definition: TypesUtils.hpp:202
DataType
Definition: Types.hpp:48

◆ RegisterDebugCallback()

void RegisterDebugCallback ( const DebugCallbackFunction func)
overridevirtual

Reimplemented from IWorkload.

Definition at line 55 of file RefDebugWorkload.cpp.

Referenced by RefDebugWorkload< DataType >::GetName().

56 {
57  m_Callback = func;
58 }

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