ArmNN
 22.05
Public Member Functions | List of all members
RefBaseWorkload< QueueDescriptor > Class Template Reference

#include <RefBaseWorkload.hpp>

Inheritance diagram for RefBaseWorkload< QueueDescriptor >:
BaseWorkload< QueueDescriptor > IWorkload

Public Member Functions

 RefBaseWorkload (const QueueDescriptor &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< QueueDescriptor >
 BaseWorkload (const QueueDescriptor &descriptor, const WorkloadInfo &info)
 
void ExecuteAsync (WorkingMemDescriptor &workingMemDescriptor) override
 
void PostAllocationConfigure () override
 
const QueueDescriptor & GetData () const
 
arm::pipe::ProfilingGuid GetGuid () const final
 
- Public Member Functions inherited from IWorkload
virtual ~IWorkload ()
 
virtual void Execute () const =0
 
virtual void RegisterDebugCallback (const DebugCallbackFunction &)
 

Additional Inherited Members

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

Detailed Description

template<typename QueueDescriptor>
class armnn::RefBaseWorkload< QueueDescriptor >

Definition at line 13 of file RefBaseWorkload.hpp.

Constructor & Destructor Documentation

◆ RefBaseWorkload()

RefBaseWorkload ( const QueueDescriptor &  descriptor,
const WorkloadInfo &  info 
)
inline

Definition at line 16 of file RefBaseWorkload.hpp.

17  : BaseWorkload<QueueDescriptor>(descriptor, info)
18  {}

Member Function Documentation

◆ ReplaceInputTensorHandle()

void ReplaceInputTensorHandle ( ITensorHandle *  tensorHandle,
unsigned int  slot 
)
inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 25 of file RefBaseWorkload.hpp.

26  {
27  this->m_Data.m_Inputs[slot] = tensorHandle;
28  }
armnn::BaseWorkload::m_Data
QueueDescriptor m_Data
Definition: Workload.hpp:81
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:26

◆ ReplaceOutputTensorHandle()

void ReplaceOutputTensorHandle ( ITensorHandle *  tensorHandle,
unsigned int  slot 
)
inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 31 of file RefBaseWorkload.hpp.

32  {
33  this->m_Data.m_Outputs[slot] = tensorHandle;
34  }
armnn::BaseWorkload::m_Data
QueueDescriptor m_Data
Definition: Workload.hpp:81
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:27

◆ SupportsTensorHandleReplacement()

virtual bool SupportsTensorHandleReplacement ( ) const
inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 20 of file RefBaseWorkload.hpp.

21  {
22  return true;
23  }
The documentation for this class was generated from the following file: