#include <ClTransposeWorkload.hpp>

Inheritance diagram for ClTransposeWorkload:

Public Member Functions
	ClTransposeWorkload (const TransposeQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)

void	Execute () const override

Public Member Functions inherited from ClBaseWorkload< TransposeQueueDescriptor >
	ClBaseWorkload (const TransposeQueueDescriptor &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< TransposeQueueDescriptor >
	BaseWorkload (const TransposeQueueDescriptor &descriptor, const WorkloadInfo &info)

void	ExecuteAsync (ExecutionData &executionData) override

void	PostAllocationConfigure () override

const TransposeQueueDescriptor &	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 ()

Static Public Member Functions
static const std::string &	GetName ()

Additional Inherited Members
Protected Member Functions inherited from ClBaseWorkload< TransposeQueueDescriptor >
virtual void	Reconfigure ()

Protected Attributes inherited from BaseWorkload< TransposeQueueDescriptor >
TransposeQueueDescriptor	m_Data

const arm::pipe::ProfilingGuid	m_Guid

Detailed Description

Definition at line 23 of file ClTransposeWorkload.hpp.

Constructor & Destructor Documentation

◆ ClTransposeWorkload()

ClTransposeWorkload	(	const TransposeQueueDescriptor &	descriptor,
		const WorkloadInfo &	info,
		const arm_compute::CLCompileContext &	clCompileContext
	)

Definition at line 29 of file ClTransposeWorkload.cpp.

References ARMNN_REPORT_PROFILING_WORKLOAD_DESC, ARMNN_SCOPED_PROFILING_EVENT, ClTransposeWorkload::GetName(), BaseWorkload< TransposeQueueDescriptor >::m_Data, TransposeDescriptor::m_DimMappings, QueueDescriptor::m_Inputs, QueueDescriptor::m_Outputs, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, armnn::Undefined, and QueueDescriptor::ValidateInputsOutputs().

Referenced by ClTransposeWorkload::GetName().

     : ClBaseWorkload<TransposeQueueDescriptor>(descriptor, info)
 {
     // Report Profiling Details
     ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClTransposeWorkload_Construct",
                                          descriptor.m_Parameters,
                                          info,
                                          this->GetGuid());
 
     m_Data.ValidateInputsOutputs(GetName(), 1, 1);
 
     const arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
     arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
     const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings;
     {
         ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClTransposeWorkload_configure");
         // Run the layer.
         m_PermuteFunction.configure(clCompileContext,
                                     &input,
                                     &output,
                                     armcomputetensorutils::BuildArmComputeTransposeVector(mappings));
     }
 }

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 55 of file ClTransposeWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT_CL_GUID, CHECK_LOCATION, BaseWorkload< TransposeQueueDescriptor >::GetGuid(), ClTransposeWorkload::GetName(), and armnn::RunClFunction().

Referenced by ClTransposeWorkload::GetName().

 {
     ARMNN_SCOPED_PROFILING_EVENT_CL_GUID(GetName() + "_Execute", this->GetGuid());
     RunClFunction(m_PermuteFunction, CHECK_LOCATION());
 }

◆ GetName()

static const std::string& GetName ( )

inlinestatic

Definition at line 26 of file ClTransposeWorkload.hpp.

References ClTransposeWorkload::ClTransposeWorkload(), ClTransposeWorkload::Execute(), and armnn::info.

Referenced by ClTransposeWorkload::ClTransposeWorkload(), and ClTransposeWorkload::Execute().

     {
         static const std::string name = std::string("ClTransposeWorkload");
         return name;
     }

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

src/backends/cl/workloads/ClTransposeWorkload.hpp
src/backends/cl/workloads/ClTransposeWorkload.cpp

Public Member Functions