#include <RefDebugWorkload.hpp>

Inheritance diagram for RefDebugWorkload< DataType >:

Public Member Functions
	RefDebugWorkload (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info)

void	Execute () const 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	PostAllocationConfigure () override

const DebugQueueDescriptor &	GetData () const

profiling::ProfilingGuid	GetGuid () const final

Public Member Functions inherited from IWorkload
virtual	~IWorkload ()

Static Public Member Functions
static const std::string &	GetName ()

Additional Inherited Members
Protected Attributes inherited from BaseWorkload< DebugQueueDescriptor >
const DebugQueueDescriptor	m_Data

const profiling::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) {}

armnn::BoostLogSeverityMapping::info

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 18 of file RefDebugWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, armnn::Debug(), TensorInfo::GetNumElements(), and armnn::GetTensorInfo().

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

 {
     using T = ResolveType<DataType>;
 
     ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, GetName() + "_Execute");
 
     const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]);
 
     const T* inputData = GetInputTensorData<T>(0, m_Data);
     T* outputData = GetOutputTensorData<T>(0, m_Data);
 
     if (m_Callback)
     {
         m_Callback(m_Data.m_Guid, m_Data.m_SlotIndex, m_Data.m_Inputs[0]);
     }
     else
     {
         Debug(inputInfo, inputData, m_Data.m_Guid, m_Data.m_LayerName, m_Data.m_SlotIndex);
     }
 
     std::memcpy(outputData, inputData, inputInfo.GetNumElements()*sizeof(T));
 }

◆ GetName()

static const std::string& GetName ( )

inlinestatic

Definition at line 23 of file RefDebugWorkload.hpp.

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

     {
         static const std::string name = std::string("RefDebug") + GetDataTypeName(DataType) + "Workload";
         return name;
     }

◆ RegisterDebugCallback()

void RegisterDebugCallback ( const DebugCallbackFunction & func )

overridevirtual

Reimplemented from IWorkload.

Definition at line 42 of file RefDebugWorkload.cpp.

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

 {
     m_Callback = func;
 }

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

src/backends/reference/workloads/RefDebugWorkload.hpp
src/backends/reference/workloads/RefDebugWorkload.cpp

Public Member Functions

Static Public Member Functions

Additional Inherited Members