RefRankWorkload Struct Reference

#include <RefRankWorkload.hpp>

Inheritance diagram for RefRankWorkload:

Public Member Functions
virtual void	Execute () const override
Public Member Functions inherited from BaseWorkload< RankQueueDescriptor >
	BaseWorkload (const RankQueueDescriptor &descriptor, const WorkloadInfo &info)
void	PostAllocationConfigure () override
const RankQueueDescriptor &	GetData () const
profiling::ProfilingGuid	GetGuid () const final
Public Member Functions inherited from IWorkload
virtual	~IWorkload ()
virtual void	RegisterDebugCallback (const DebugCallbackFunction &)

Additional Inherited Members
Protected Attributes inherited from BaseWorkload< RankQueueDescriptor >
const RankQueueDescriptor	m_Data
const profiling::ProfilingGuid	m_Guid

Detailed Description

Definition at line 16 of file RefRankWorkload.hpp.

Member Function Documentation

Execute()

virtual void Execute ( ) const

inlineoverridevirtual

Implements IWorkload.

Definition at line 20 of file RefRankWorkload.hpp.

References TensorInfo::GetNumDimensions(), armnn::GetTensorInfo(), BaseWorkload< RankQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, and QueueDescriptor::m_Outputs.

    {
        const int32_t rank = static_cast<int32_t>(GetTensorInfo(m_Data.m_Inputs[0]).GetNumDimensions());

        std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t));
        m_Data.m_Outputs[0]->Unmap();
    }

---

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

• src/backends/reference/workloads/RefRankWorkload.hpp