#include <NeonBaseWorkload.hpp>

Inheritance diagram for NeonBaseWorkload< QueueDescriptor >:

Collaboration diagram for NeonBaseWorkload< QueueDescriptor >:

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

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	ExecuteAsync (ExecutionData &executionData) override

void	PostAllocationConfigure () override

const QueueDescriptor &	GetData () const

arm::pipe::ProfilingGuid	GetGuid () const final

virtual bool	SupportsTensorHandleReplacement () const 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 Member Functions
virtual void	Reconfigure ()

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::NeonBaseWorkload< QueueDescriptor >

Definition at line 13 of file NeonBaseWorkload.hpp.

Constructor & Destructor Documentation

◆ NeonBaseWorkload()

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

inline

Definition at line 16 of file NeonBaseWorkload.hpp.

17 : BaseWorkload<QueueDescriptor>(descriptor, info)

18 {}

Member Function Documentation

◆ Reconfigure()

virtual void Reconfigure ( )

inlineprotectedvirtual

Definition at line 56 of file NeonBaseWorkload.hpp.

     {
         throw armnn::UnimplementedException("Reconfigure not implemented for this workload");
     }

Referenced by NeonBaseWorkload< RsqrtQueueDescriptor >::ReplaceInputTensorHandle(), and NeonBaseWorkload< RsqrtQueueDescriptor >::ReplaceOutputTensorHandle().

◆ ReplaceInputTensorHandle()

void ReplaceInputTensorHandle	(	ITensorHandle *	tensorHandle,
		unsigned int	slot
	)

inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 21 of file NeonBaseWorkload.hpp.

     {
         ITensorHandle* backupHandle = this->m_Data.m_Inputs[slot];
         this->m_Data.m_Inputs[slot] = tensorHandle;
         try
         {
             Reconfigure();
         }
         catch(armnn::UnimplementedException& e)
         {
             // Cannot reconfigure, revert the slot back and throw the exception.
             this->m_Data.m_Inputs[slot] = backupHandle;
             throw e;
         }
     }

◆ ReplaceOutputTensorHandle()

void ReplaceOutputTensorHandle	(	ITensorHandle *	tensorHandle,
		unsigned int	slot
	)

inlineoverridevirtual

Reimplemented from BaseWorkload< QueueDescriptor >.

Definition at line 38 of file NeonBaseWorkload.hpp.

     {
         ITensorHandle* backupHandle = this->m_Data.m_Outputs[slot];
         this->m_Data.m_Outputs[slot] = tensorHandle;
         try
         {
             Reconfigure();
         }
         catch(armnn::UnimplementedException& e)
         {
             // Cannot reconfigure, revert the slot back and throw the exception.
             this->m_Data.m_Inputs[slot] = backupHandle;
             throw e;
         }
     }

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

src/backends/neon/workloads/NeonBaseWorkload.hpp

Public Member Functions

Protected Member Functions

Additional Inherited Members