#include <ClPermuteWorkload.hpp>

Inheritance diagram for ClPermuteWorkload:

Public Member Functions
	ClPermuteWorkload (const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)

void	Execute () const override

Public Member Functions inherited from BaseWorkload< PermuteQueueDescriptor >
	BaseWorkload (const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info)

void	ExecuteAsync (WorkingMemDescriptor &workingMemDescriptor) override

void	PostAllocationConfigure () override

const PermuteQueueDescriptor &	GetData () const

profiling::ProfilingGuid	GetGuid () const final

Public Member Functions inherited from IWorkload
virtual	~IWorkload ()

virtual void	RegisterDebugCallback (const DebugCallbackFunction &)

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

Additional Inherited Members
Protected Attributes inherited from BaseWorkload< PermuteQueueDescriptor >
PermuteQueueDescriptor	m_Data

const profiling::ProfilingGuid	m_Guid

Detailed Description

Definition at line 23 of file ClPermuteWorkload.hpp.

Constructor & Destructor Documentation

◆ ClPermuteWorkload()

ClPermuteWorkload	(	const PermuteQueueDescriptor &	descriptor,
		const WorkloadInfo &	info,
		const arm_compute::CLCompileContext &	clCompileContext
	)

Definition at line 29 of file ClPermuteWorkload.cpp.

References ARMNN_REPORT_PROFILING_WORKLOAD_DESC, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by ClPermuteWorkload::GetName().

     : BaseWorkload<PermuteQueueDescriptor>(descriptor, info)
 {
     // Report Profiling Details
     ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClPermuteWorkload_Construct",
                                          descriptor.m_Parameters,
                                          info,
                                          this->GetGuid());
 
     using armcomputetensorutils::BuildArmComputePermutationVector;
 
     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, "ClPermuteWorkload_configure");
         // Run the layer.
         m_PermuteFunction.configure(clCompileContext, &input, &output, BuildArmComputePermutationVector(mappings));
     }
 }

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 55 of file ClPermuteWorkload.cpp.

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

Referenced by ClPermuteWorkload::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 ClPermuteWorkload.hpp.

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

Referenced by ClPermuteWorkload::Execute().

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

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

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

Public Member Functions