#include <Workload.hpp>

Inheritance diagram for BaseWorkload< QueueDescriptor >:

Collaboration diagram for BaseWorkload< QueueDescriptor >:

Public Member Functions
	BaseWorkload (const QueueDescriptor &descriptor, const WorkloadInfo &info)

virtual const std::string &	GetName () const override

void	ExecuteAsync (ExecutionData &executionData) override

void	PostAllocationConfigure () override

const QueueDescriptor &	GetData () 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 void	Execute () const =0

virtual void	RegisterDebugCallback (const DebugCallbackFunction &)

virtual armnn::Optional< armnn::MemoryRequirements >	GetMemoryRequirements ()

Protected Attributes
QueueDescriptor	m_Data

const arm::pipe::ProfilingGuid	m_Guid

const std::string	m_Name

Detailed Description

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

Definition at line 33 of file Workload.hpp.

Constructor & Destructor Documentation

◆ BaseWorkload()

BaseWorkload	(	const QueueDescriptor &	descriptor,
		const WorkloadInfo &	info
	)

inline

Definition at line 37 of file Workload.hpp.

         : m_Data(descriptor),
           m_Guid(arm::pipe::IProfilingService::GetNextGuid()),
           m_Name(info.m_Name)
     {
         m_Data.Validate(info);
     }

Member Function Documentation

◆ ExecuteAsync()

void ExecuteAsync ( ExecutionData & executionData )

inlineoverridevirtual

Implements IWorkload.

Reimplemented in RefConvertFp16ToFp32Workload, RefConvertFp32ToFp16Workload, and RefFakeQuantizationFloat32Workload.

Definition at line 50 of file Workload.hpp.

     {
         ARMNN_LOG(info) << "Using default async workload execution, this will network affect performance";
 #if !defined(ARMNN_DISABLE_THREADS)
         std::lock_guard<std::mutex> lockGuard(m_AsyncWorkloadMutex);
 #endif
         WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
         m_Data.m_Inputs = workingMemDescriptor->m_Inputs;
         m_Data.m_Outputs = workingMemDescriptor->m_Outputs;
  
         Execute();
     };

◆ GetData()

const QueueDescriptor& GetData ( ) const

inline

Definition at line 65 of file Workload.hpp.

65 { return m_Data; }

◆ GetGuid()

arm::pipe::ProfilingGuid GetGuid ( ) const

inlinefinalvirtual

Implements IWorkload.

Definition at line 67 of file Workload.hpp.

67 { return m_Guid; }

Referenced by ClLstmFloatWorkload::ClLstmFloatWorkload(), and NeonLstmFloatWorkload::NeonLstmFloatWorkload().

◆ GetName()

virtual const std::string& GetName ( ) const

inlineoverridevirtual

Implements IWorkload.

Definition at line 45 of file Workload.hpp.

     {
         return m_Name;
     }

◆ PostAllocationConfigure()

void PostAllocationConfigure ( )

inlineoverridevirtual

Implements IWorkload.

Definition at line 63 of file Workload.hpp.

63 {}

◆ ReplaceInputTensorHandle()

void ReplaceInputTensorHandle	(	ITensorHandle *	tensorHandle,
		unsigned int	slot
	)

inlineoverridevirtual

Implements IWorkload.

Definition at line 75 of file Workload.hpp.

     {
         armnn::IgnoreUnused(tensorHandle, slot);
         throw armnn::UnimplementedException("ReplaceInputTensorHandle not implemented for this workload");
     }

◆ ReplaceOutputTensorHandle()

void ReplaceOutputTensorHandle	(	ITensorHandle *	tensorHandle,
		unsigned int	slot
	)

inlineoverridevirtual

Implements IWorkload.

Definition at line 82 of file Workload.hpp.

     {
         armnn::IgnoreUnused(tensorHandle, slot);
         throw armnn::UnimplementedException("ReplaceOutputTensorHandle not implemented for this workload");
     }

◆ SupportsTensorHandleReplacement()

virtual bool SupportsTensorHandleReplacement ( ) const

inlineoverridevirtual

Implements IWorkload.

Definition at line 69 of file Workload.hpp.

     {
         return false;
     }

Member Data Documentation

◆ m_Data

QueueDescriptor m_Data

protected

Definition at line 89 of file Workload.hpp.

◆ m_Guid

const arm::pipe::ProfilingGuid m_Guid

protected

Definition at line 90 of file Workload.hpp.

Referenced by BaseWorkload< SplitterQueueDescriptor >::GetGuid().

◆ m_Name

const std::string m_Name

protected

Definition at line 91 of file Workload.hpp.

Referenced by BaseWorkload< SplitterQueueDescriptor >::GetName().

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

include/armnn/backends/Workload.hpp

Public Member Functions

Protected Attributes

Detailed Description

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

Constructor & Destructor Documentation

◆ BaseWorkload()

Member Function Documentation

◆ ExecuteAsync()

◆ GetData()

◆ GetGuid()

◆ GetName()

◆ PostAllocationConfigure()

◆ ReplaceInputTensorHandle()

◆ ReplaceOutputTensorHandle()

◆ SupportsTensorHandleReplacement()

Member Data Documentation

◆ m_Data

◆ m_Guid

◆ m_Name

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