#include <NeonPermuteWorkload.hpp>

Inheritance diagram for NeonPermuteWorkload:

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

void	Execute () const override

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

void	ExecuteAsync (ExecutionData &executionData) override

void	PostAllocationConfigure () override

const PermuteQueueDescriptor &	GetData () const

arm::pipe::ProfilingGuid	GetGuid () const final

virtual bool	SupportsTensorHandleReplacement () const override

Public Member Functions inherited from IWorkload
virtual	~IWorkload ()

virtual arm::pipe::ProfilingGuid	GetGuid () const =0

virtual bool	SupportsTensorHandleReplacement () const =0

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 NeonBaseWorkload< PermuteQueueDescriptor >
virtual void	Reconfigure ()

Protected Attributes inherited from BaseWorkload< PermuteQueueDescriptor >
PermuteQueueDescriptor	m_Data

const arm::pipe::ProfilingGuid	m_Guid

Detailed Description

Definition at line 22 of file NeonPermuteWorkload.hpp.

Constructor & Destructor Documentation

◆ NeonPermuteWorkload()

NeonPermuteWorkload	(	const PermuteQueueDescriptor &	descriptor,
		const WorkloadInfo &	info
	)

Definition at line 27 of file NeonPermuteWorkload.cpp.

         : NeonBaseWorkload<PermuteQueueDescriptor>(descriptor, info)
 {
     // Report Profiling Details
     ARMNN_REPORT_PROFILING_WORKLOAD_DESC("NeonPermuteWorkload_Construct",
                                          descriptor.m_Parameters,
                                          info,
                                          this->GetGuid());
  
     using armcomputetensorutils::BuildArmComputePermutationVector;
  
     m_Data.ValidateInputsOutputs(GetName(), 1, 1);
  
     const arm_compute::ITensor& input = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
     arm_compute::ITensor& output = static_cast<IAclTensorHandle*>(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));
 }

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

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 49 of file NeonPermuteWorkload.cpp.

 {
     ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID(GetName() + "_Execute", this->GetGuid());
     m_PermuteFunction.run();
 }

References ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID, BaseWorkload< PermuteQueueDescriptor >::GetGuid(), and NeonPermuteWorkload::GetName().

◆ GetName()

static const std::string& GetName ( )

inlinestatic

Definition at line 25 of file NeonPermuteWorkload.hpp.

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

Referenced by NeonPermuteWorkload::Execute().

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

src/backends/neon/workloads/NeonPermuteWorkload.hpp
src/backends/neon/workloads/NeonPermuteWorkload.cpp

Public Member Functions