NeonTransposeConvolution2dWorkload Class Reference

#include <NeonTransposeConvolution2dWorkload.hpp>

Inheritance diagram for NeonTransposeConvolution2dWorkload:

Public Member Functions
	NeonTransposeConvolution2dWorkload (const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
void	Execute () const override
Public Member Functions inherited from BaseWorkload< TransposeConvolution2dQueueDescriptor >
	BaseWorkload (const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
void	PostAllocationConfigure () override
const TransposeConvolution2dQueueDescriptor &	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< TransposeConvolution2dQueueDescriptor >
const TransposeConvolution2dQueueDescriptor	m_Data
const profiling::ProfilingGuid	m_Guid

Detailed Description

Definition at line 27 of file NeonTransposeConvolution2dWorkload.hpp.

Constructor & Destructor Documentation

NeonTransposeConvolution2dWorkload()

NeonTransposeConvolution2dWorkload	(	const TransposeConvolution2dQueueDescriptor &	descriptor,
		const WorkloadInfo &	info,
		std::shared_ptr< arm_compute::MemoryManagerOnDemand > &	memoryManager
	)

Definition at line 55 of file NeonTransposeConvolution2dWorkload.cpp.

References BaseWorkload< TransposeConvolution2dQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, QueueDescriptor::m_Outputs, and QueueDescriptor::ValidateInputsOutputs().

    : BaseWorkload<TransposeConvolution2dQueueDescriptor>(descriptor, info)
{
    m_Data.ValidateInputsOutputs("NeonTransposeConvolution2dWorkload", 1, 1);

    arm_compute::ITensor& input  = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
    arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();

    arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
    input.info()->set_data_layout(aclDataLayout);
    output.info()->set_data_layout(aclDataLayout);

    m_KernelTensor = std::make_unique<arm_compute::Tensor>();
    BuildArmComputeTensor(*m_KernelTensor, m_Data.m_Weight->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout);

    if (m_Data.m_Parameters.m_BiasEnabled)
    {
        m_BiasTensor = std::make_unique<arm_compute::Tensor>();
        BuildArmComputeTensor(*m_BiasTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout);
    }

    arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(m_Data.m_Parameters);

    m_Layer = std::make_unique<arm_compute::NEDeconvolutionLayer>(memoryManager);
    m_Layer->configure(&input, m_KernelTensor.get(), m_BiasTensor.get(), &output, padStrideInfo);

    ARMNN_ASSERT(m_Layer);

    InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight);

    if (m_Data.m_Parameters.m_BiasEnabled)
    {
        InitializeArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias);
    }

    m_Layer->prepare();
    FreeUnusedTensors();
}

Member Function Documentation

Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 96 of file NeonTransposeConvolution2dWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT_NEON.

{
    ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonTransposeConvolution2dWorkload_Execute");
    m_Layer->run();
}

---

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

• src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.hpp
• src/backends/neon/workloads/NeonTransposeConvolution2dWorkload.cpp