NeonRankWorkload Struct Reference

#include <NeonRankWorkload.hpp>

Inheritance diagram for NeonRankWorkload:

Collaboration diagram for NeonRankWorkload:

Public Member Functions
virtual void	Execute () const override
Public Member Functions inherited from NeonBaseWorkload< RankQueueDescriptor >
	NeonBaseWorkload (const RankQueueDescriptor &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	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	RegisterDebugCallback (const DebugCallbackFunction &)
virtual armnn::Optional< armnn::MemoryRequirements >	GetMemoryRequirements ()

Additional Inherited Members
Protected Member Functions inherited from NeonBaseWorkload< RankQueueDescriptor >
virtual void	Reconfigure ()
Protected Attributes inherited from BaseWorkload< QueueDescriptor >
QueueDescriptor	m_Data
const arm::pipe::ProfilingGuid	m_Guid
const std::string	m_Name

Detailed Description

Definition at line 16 of file NeonRankWorkload.hpp.

Member Function Documentation

Execute()

virtual void Execute ( ) const

inlineoverridevirtual

Implements IWorkload.

Definition at line 20 of file NeonRankWorkload.hpp.

    {
        ARMNN_SCOPED_PROFILING_EVENT_NEON_NAME_GUID("NeonRankWorkload_Execute");
 
        const NeonTensorHandle* neonTensorHandle = PolymorphicDowncast<const NeonTensorHandle*>(m_Data.m_Inputs[0]);
        const int32_t rank = static_cast<int32_t>(neonTensorHandle->GetShape().GetNumDimensions());
 
        std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t));
        m_Data.m_Outputs[0]->Unmap();
    }

References ARMNN_SCOPED_PROFILING_EVENT_NEON_NAME_GUID, TensorShape::GetNumDimensions(), NeonTensorHandle::GetShape(), BaseWorkload< QueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, and QueueDescriptor::m_Outputs.

---

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

• src/backends/neon/workloads/NeonRankWorkload.hpp