ArmNN
 26.01
Loading...
Searching...
No Matches
TosaRefBaseWorkload< QueueDescriptor > Class Template Reference

#include <TosaRefBaseWorkload.hpp>

Inheritance diagram for TosaRefBaseWorkload< QueueDescriptor >:
[legend]
Collaboration diagram for TosaRefBaseWorkload< QueueDescriptor >:
[legend]

Public Member Functions

 TosaRefBaseWorkload (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)
 
virtual const std::string & GetName () const override
 
void PostAllocationConfigure () override
 
const QueueDescriptorGetData () 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 &)
 
virtual armnn::Optional< armnn::MemoryRequirementsGetMemoryRequirements ()
 

Additional Inherited Members

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

Detailed Description

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

Definition at line 14 of file TosaRefBaseWorkload.hpp.

Constructor & Destructor Documentation

◆ TosaRefBaseWorkload()

template<typename QueueDescriptor >
TosaRefBaseWorkload ( const QueueDescriptor & descriptor,
const WorkloadInfo & info )
inline

Definition at line 17 of file TosaRefBaseWorkload.hpp.

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

Member Function Documentation

◆ ReplaceInputTensorHandle()

template<typename QueueDescriptor >
void ReplaceInputTensorHandle ( ITensorHandle * tensorHandle,
unsigned int slot )
inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 27 of file TosaRefBaseWorkload.hpp.

28 {
29 this->m_Data.m_Inputs[slot] = tensorHandle;
30 }
QueueDescriptor m_Data
Definition Workload.hpp:74
std::vector< ITensorHandle * > m_Inputs

References BaseWorkload< QueueDescriptor >::m_Data, and QueueDescriptor::m_Inputs.

◆ ReplaceOutputTensorHandle()

template<typename QueueDescriptor >
void ReplaceOutputTensorHandle ( ITensorHandle * tensorHandle,
unsigned int slot )
inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 33 of file TosaRefBaseWorkload.hpp.

34 {
35 this->m_Data.m_Outputs[slot] = tensorHandle;
36 }
std::vector< ITensorHandle * > m_Outputs

References BaseWorkload< QueueDescriptor >::m_Data, and QueueDescriptor::m_Outputs.

◆ SupportsTensorHandleReplacement()

template<typename QueueDescriptor >
virtual bool SupportsTensorHandleReplacement ( ) const
inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 21 of file TosaRefBaseWorkload.hpp.

22 {
23 return true;
24 }

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