#include <ClPermuteWorkload.hpp>

Inheritance diagram for ClPermuteWorkload:

Public Member Functions
	ClPermuteWorkload (const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info)

void	Execute () const override

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

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 >
const 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
	)

Definition at line 29 of file ClPermuteWorkload.cpp.

Referenced by ClPermuteWorkload::GetName().

     : BaseWorkload<PermuteQueueDescriptor>(descriptor, info)
 {
     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;
 
     // Run the layer.
     m_PermuteFunction.configure(&input, &output, BuildArmComputePermutationVector(mappings));
 }

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 45 of file ClPermuteWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT_CL, CHECK_LOCATION, ClPermuteWorkload::GetName(), and armnn::RunClFunction().

Referenced by ClPermuteWorkload::GetName().

 {
     ARMNN_SCOPED_PROFILING_EVENT_CL( GetName() + "_Execute");
     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