From fd627ffaec8fd8801d980b4c91ee7c0607ab6aaf Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Thu, 25 Feb 2021 17:44:00 +0000 Subject: IVGCVSW-5687 Update Doxygen Docu * Update Doxygen Documentation for 21.02 release Signed-off-by: Jan Eilers Change-Id: I9ed2f9caab038836ea99d7b378d7899fe431a4e5 --- ...lassarmnn_1_1_neon_convolution2d_workload.xhtml | 281 +++++++++++++++++++++ 1 file changed, 281 insertions(+) create mode 100644 21.02/classarmnn_1_1_neon_convolution2d_workload.xhtml (limited to '21.02/classarmnn_1_1_neon_convolution2d_workload.xhtml') diff --git a/21.02/classarmnn_1_1_neon_convolution2d_workload.xhtml b/21.02/classarmnn_1_1_neon_convolution2d_workload.xhtml new file mode 100644 index 0000000000..3ac5f7acea --- /dev/null +++ b/21.02/classarmnn_1_1_neon_convolution2d_workload.xhtml @@ -0,0 +1,281 @@ + + + + + + + + + + + + + +ArmNN: NeonConvolution2dWorkload Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Public Member Functions | +List of all members
+
+
NeonConvolution2dWorkload Class Reference
+
+
+ +

#include <NeonConvolution2dWorkload.hpp>

+
+Inheritance diagram for NeonConvolution2dWorkload:
+
+
+ + +BaseWorkload< Convolution2dQueueDescriptor > +IWorkload + +
+ + + + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 NeonConvolution2dWorkload (const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager, const bool isFastMathENabled=false)
 
void Execute () const override
 
arm_compute::ConvolutionMethod GetConvolutionMethod () const
 
- Public Member Functions inherited from BaseWorkload< Convolution2dQueueDescriptor >
 BaseWorkload (const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void PostAllocationConfigure () override
 
const Convolution2dQueueDescriptor & 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< Convolution2dQueueDescriptor >
const Convolution2dQueueDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 
+

Detailed Description

+
+

Definition at line 27 of file NeonConvolution2dWorkload.hpp.

+

Constructor & Destructor Documentation

+ +

◆ NeonConvolution2dWorkload()

+ +
+
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
NeonConvolution2dWorkload (const Convolution2dQueueDescriptor & descriptor,
const WorkloadInfo & info,
std::shared_ptr< arm_compute::MemoryManagerOnDemand > & memoryManager,
const bool isFastMathENabled = false 
)
+
+ +

Definition at line 66 of file NeonConvolution2dWorkload.cpp.

+ +

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

+
71  : BaseWorkload<Convolution2dQueueDescriptor>(descriptor, info)
72 {
73  using arm_compute::NEConvolutionLayer;
74 
75  m_Data.ValidateInputsOutputs("NeonConvolution2dWorkload", 1, 1);
76 
77  // todo: check tensor shapes match.
78 
79  arm_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
80  arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
81 
82  arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
83  input.info()->set_data_layout(aclDataLayout);
84  output.info()->set_data_layout(aclDataLayout);
85 
86  m_KernelTensor = std::make_unique<arm_compute::Tensor>();
87  BuildArmComputeTensor(*m_KernelTensor, m_Data.m_Weight->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout);
88 
89  if (m_Data.m_Parameters.m_BiasEnabled)
90  {
91  m_BiasTensor = std::make_unique<arm_compute::Tensor>();
92  BuildArmComputeTensor(*m_BiasTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout);
93  }
94 
95  arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(m_Data.m_Parameters);
96 
97  const arm_compute::Size2D aclDilationInfo = BuildArmComputeSize2D(m_Data.m_Parameters.m_DilationX,
98  m_Data.m_Parameters.m_DilationY);
99 
100  const arm_compute::ActivationLayerInfo activationInfo = ConvertAdditionalInfoToAclActivationLayerInfo(descriptor);
101 
102  auto convolutionLayer = std::make_unique<arm_compute::NEConvolutionLayer>(memoryManager);
103  convolutionLayer->configure(&input,
104  m_KernelTensor.get(),
105  m_BiasTensor.get(),
106  &output,
107  padStrideInfo,
108  arm_compute::WeightsInfo(),
109  aclDilationInfo,
110  activationInfo,
111  isFastMathEnabled);
112 
113  m_ConvolutionMethod =
114  convolutionLayer->get_convolution_method(input.info(),
115  m_KernelTensor->info(),
116  output.info(),
117  padStrideInfo,
118  arm_compute::WeightsInfo(),
119  aclDilationInfo,
120  activationInfo,
121  isFastMathEnabled);
122 
123  m_ConvolutionLayer.reset(convolutionLayer.release());
124 
125  ARMNN_ASSERT(m_ConvolutionLayer);
126 
127  InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight);
128 
129  if (m_Data.m_Parameters.m_BiasEnabled)
130  {
131  InitializeArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias);
132  }
133 
134  m_ConvolutionLayer->prepare();
135  FreeUnusedTensors();
136 }
armnn::Convolution2dDescriptor::m_BiasEnabled
bool m_BiasEnabled
Enable/disable bias.
Definition: Descriptors.hpp:454
+
armnn::Convolution2dQueueDescriptor::m_Bias
const ConstCpuTensorHandle * m_Bias
Definition: WorkloadData.hpp:201
+
armnn::Convolution2dDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:456
+
armnn::DataLayout
DataLayout
Definition: Types.hpp:50
+
armnn::BaseWorkload< Convolution2dQueueDescriptor >::m_Data
const Convolution2dQueueDescriptor m_Data
Definition: Workload.hpp:46
+
armnn::ConvertAdditionalInfoToAclActivationLayerInfo
arm_compute::ActivationLayerInfo ConvertAdditionalInfoToAclActivationLayerInfo(const QueueDescriptor &queueDescriptor)
Definition: ArmComputeUtils.hpp:93
+
armnn::QueueDescriptor::ValidateInputsOutputs
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Definition: WorkloadData.cpp:460
+
armnn::Convolution2dDescriptor::m_DilationY
uint32_t m_DilationY
Dilation along y axis.
Definition: Descriptors.hpp:452
+ +
armnn::Convolution2dQueueDescriptor::m_Weight
const ConstCpuTensorHandle * m_Weight
Definition: WorkloadData.hpp:200
+
ARMNN_ASSERT
#define ARMNN_ASSERT(COND)
Definition: Assert.hpp:14
+
armnn::InitializeArmComputeTensorData
void InitializeArmComputeTensorData(arm_compute::Tensor &tensor, const ConstCpuTensorHandle *handle)
Definition: NeonWorkloadUtils.hpp:35
+
armnn::Convolution2dDescriptor::m_DilationX
uint32_t m_DilationX
Dilation along x axis.
Definition: Descriptors.hpp:450
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+ +
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
armnn::ConstCpuTensorHandle::GetTensorInfo
const TensorInfo & GetTensorInfo() const
Definition: CpuTensorHandle.hpp:37
+
+
+
+

Member Function Documentation

+ +

◆ Execute()

+ +
+
+ + + + + +
+ + + + + + + +
void Execute () const
+ 		+overridevirtual
+
+ +

Implements IWorkload.

+ +

Definition at line 138 of file NeonConvolution2dWorkload.cpp.

+ +

References ARMNN_SCOPED_PROFILING_EVENT_NEON.

+
139 {
140  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonConvolution2dWorkload_Execute");
141  m_ConvolutionLayer->run();
142 }
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+
+
+
+ +

◆ GetConvolutionMethod()

+ +
+
+ + + + + + + +
arm_compute::ConvolutionMethod GetConvolutionMethod () const
+
+ +

Definition at line 144 of file NeonConvolution2dWorkload.cpp.

+
145 {
146  return m_ConvolutionMethod;
147 }
+
+
+
The documentation for this class was generated from the following files: +
+
+ + + + -- cgit v1.2.1