ArmNN
 22.11
Public Member Functions | List of all members
NeonBatchToSpaceNdWorkload Class Reference

#include <NeonBatchToSpaceNdWorkload.hpp>

Inheritance diagram for NeonBatchToSpaceNdWorkload:
NeonBaseWorkload< BatchToSpaceNdQueueDescriptor > BaseWorkload< BatchToSpaceNdQueueDescriptor > IWorkload

Public Member Functions

 NeonBatchToSpaceNdWorkload (const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info)
 
virtual void Execute () const override
 
- Public Member Functions inherited from NeonBaseWorkload< BatchToSpaceNdQueueDescriptor >
 NeonBaseWorkload (const BatchToSpaceNdQueueDescriptor &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< BatchToSpaceNdQueueDescriptor >
 BaseWorkload (const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void ExecuteAsync (ExecutionData &executionData) override
 
void PostAllocationConfigure () override
 
const BatchToSpaceNdQueueDescriptor & 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 ()
 

Additional Inherited Members

- Protected Member Functions inherited from NeonBaseWorkload< BatchToSpaceNdQueueDescriptor >
virtual void Reconfigure ()
 
- Protected Attributes inherited from BaseWorkload< BatchToSpaceNdQueueDescriptor >
BatchToSpaceNdQueueDescriptor m_Data
 
const arm::pipe::ProfilingGuid m_Guid
 

Detailed Description

Definition at line 22 of file NeonBatchToSpaceNdWorkload.hpp.

Constructor & Destructor Documentation

◆ NeonBatchToSpaceNdWorkload()

NeonBatchToSpaceNdWorkload ( const BatchToSpaceNdQueueDescriptor &  descriptor,
const WorkloadInfo &  info 
)

Definition at line 38 of file NeonBatchToSpaceNdWorkload.cpp.

References ARMNN_REPORT_PROFILING_WORKLOAD_DESC, BaseWorkload< BatchToSpaceNdQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, QueueDescriptor::m_Outputs, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and QueueDescriptor::ValidateInputsOutputs().

40  : NeonBaseWorkload<BatchToSpaceNdQueueDescriptor>(descriptor, info)
41 {
42  // Report Profiling Details
43  ARMNN_REPORT_PROFILING_WORKLOAD_DESC("NeonBatchToSpaceWorkload_Construct",
44  descriptor.m_Parameters,
45  info,
46  this->GetGuid());
47 
48  m_Data.ValidateInputsOutputs("NeonBatchToSpaceNdWorkload", 1, 1);
49 
50  arm_compute::ITensor& input =
51  armnn::PolymorphicPointerDowncast<IAclTensorHandle>(m_Data.m_Inputs[0])->GetTensor();
52  arm_compute::ITensor& output =
53  armnn::PolymorphicPointerDowncast<IAclTensorHandle>(m_Data.m_Outputs[0])->GetTensor();
54 
55  arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
56  input.info()->set_data_layout(aclDataLayout);
57  output.info()->set_data_layout(aclDataLayout);
58 
59  // ArmNN blockShape is [H, W] Cl asks for W, H
60  int32_t blockHeight = armnn::numeric_cast<int32_t>(descriptor.m_Parameters.m_BlockShape[0]);
61  int32_t blockWidth = armnn::numeric_cast<int32_t>(descriptor.m_Parameters.m_BlockShape[1]);
62 
63  m_Layer.reset(new arm_compute::NEBatchToSpaceLayer());
64  m_Layer->configure(&input, blockWidth, blockHeight, &output);
65  m_Layer->prepare();
66 }
armnn::DataLayout
DataLayout
Definition: Types.hpp:62
armnn::BatchToSpaceNdDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:863
armnn::QueueDescriptor::ValidateInputsOutputs
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Definition: WorkloadData.cpp:475
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:27
ARMNN_REPORT_PROFILING_WORKLOAD_DESC
#define ARMNN_REPORT_PROFILING_WORKLOAD_DESC(name, desc, infos, guid)
Definition: Profiling.hpp:227
armnn::numeric_cast
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
Definition: NumericCast.hpp:35
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:26

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 68 of file NeonBatchToSpaceNdWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID, and BaseWorkload< BatchToSpaceNdQueueDescriptor >::GetGuid().

69 {
70  if (m_Layer)
71  {
72  ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID("NeonSpaceToBatchNdWorkload_Execute", this->GetGuid());
73  m_Layer->run();
74  }
75 }
armnn::BaseWorkload< BatchToSpaceNdQueueDescriptor >::GetGuid
arm::pipe::ProfilingGuid GetGuid() const final
Definition: Workload.hpp:61
ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID
#define ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID(name, guid)
Definition: NeonWorkloadUtils.hpp:24
The documentation for this class was generated from the following files: