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 --- ...n_1_1_neon_depthwise_convolution_workload.xhtml | 256 +++++++++++++++++++++ 1 file changed, 256 insertions(+) create mode 100644 21.02/classarmnn_1_1_neon_depthwise_convolution_workload.xhtml (limited to '21.02/classarmnn_1_1_neon_depthwise_convolution_workload.xhtml') diff --git a/21.02/classarmnn_1_1_neon_depthwise_convolution_workload.xhtml b/21.02/classarmnn_1_1_neon_depthwise_convolution_workload.xhtml new file mode 100644 index 0000000000..918fda57c8 --- /dev/null +++ b/21.02/classarmnn_1_1_neon_depthwise_convolution_workload.xhtml @@ -0,0 +1,256 @@ + + + + + + + + + + + + + +ArmNN: NeonDepthwiseConvolutionWorkload Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Public Member Functions | +List of all members
+
+
NeonDepthwiseConvolutionWorkload Class Reference
+
+
+ +

#include <NeonDepthwiseConvolutionWorkload.hpp>

+
+Inheritance diagram for NeonDepthwiseConvolutionWorkload:
+
+
+ + +BaseWorkload< DepthwiseConvolution2dQueueDescriptor > +IWorkload + +
+ + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 NeonDepthwiseConvolutionWorkload (const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
 
virtual void Execute () const override
 
- Public Member Functions inherited from BaseWorkload< DepthwiseConvolution2dQueueDescriptor >
 BaseWorkload (const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void PostAllocationConfigure () override
 
const DepthwiseConvolution2dQueueDescriptor & 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< DepthwiseConvolution2dQueueDescriptor >
const DepthwiseConvolution2dQueueDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 
+

Detailed Description

+
+

Definition at line 26 of file NeonDepthwiseConvolutionWorkload.hpp.

+

Constructor & Destructor Documentation

+ +

◆ NeonDepthwiseConvolutionWorkload()

+ +
+
+ + + + + + + + + + + + + + + + + + +
NeonDepthwiseConvolutionWorkload (const DepthwiseConvolution2dQueueDescriptor & descriptor,
const WorkloadInfo & info 
)
+
+ +

Definition at line 77 of file NeonDepthwiseConvolutionWorkload.cpp.

+ +

References armnn::ConvertWeightTensorFromArmnnToAcl(), TensorInfo::GetNumBytes(), ConstCpuTensorHandle::GetTensorInfo(), BaseWorkload< DepthwiseConvolution2dQueueDescriptor >::m_Data, DepthwiseConvolution2dDescriptor::m_DataLayout, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and DepthwiseConvolution2dQueueDescriptor::m_Weight.

+
80  : BaseWorkload<DepthwiseConvolution2dQueueDescriptor>(descriptor, info)
81 {
82  // ArmNN's weight format is [ M, I, H, W ]
83  auto& weightInfo = m_Data.m_Weight->GetTensorInfo();
84 
85  // Allocate a buffer for the swizzling of the weight tensor
86  std::unique_ptr<unsigned char[]> permuteBuffer(new unsigned char[m_Data.m_Weight->GetTensorInfo().GetNumBytes()]);
87 
88  // Convert the weight format from ArmNN's [ M, I, H, W ] (does NOT depend on the data layout) to either
89  // [ 1, H, W, I * M ] (if NHWC) or [ 1, I * M, H, W ] (if NCHW), as required by the compute library
90  ConstTensor weightPermuted = ConvertWeightTensorFromArmnnToAcl(m_Data.m_Weight,
91  m_Data.m_Parameters.m_DataLayout,
92  permuteBuffer.get());
93 
94  // Convert the weights into the compute library format
95  m_KernelTensor = std::make_unique<arm_compute::Tensor>();
96  BuildArmComputeTensor(*m_KernelTensor, weightPermuted.GetInfo(), m_Data.m_Parameters.m_DataLayout);
97 
98  if (m_Data.m_Parameters.m_BiasEnabled)
99  {
100  m_BiasTensor = std::make_unique<arm_compute::Tensor>();
101  BuildArmComputeTensor(*m_BiasTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout);
102  }
103 
104  const arm_compute::Size2D aclDilationInfo = BuildArmComputeSize2D(
105  m_Data.m_Parameters.m_DilationX, m_Data.m_Parameters.m_DilationY);
106 
107  m_Data.ValidateInputsOutputs("NeonDepthwiseConvolutionWorkload", 1, 1);
108 
109  IAclTensorHandle* inputTensorHandle = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0]);
110  IAclTensorHandle* outputTensorHandle = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[0]);
111 
112  arm_compute::ITensor& input = inputTensorHandle->GetTensor();
113  arm_compute::ITensor& output = outputTensorHandle->GetTensor();
114 
115  arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
116  input.info()->set_data_layout(aclDataLayout);
117  output.info()->set_data_layout(aclDataLayout);
118 
119  // Get the depth multiplier
120  const unsigned int depthMultiplier = weightInfo.GetShape()[0];
121 
122  arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(m_Data.m_Parameters);
123 
124  const arm_compute::ActivationLayerInfo activationInfo = ConvertAdditionalInfoToAclActivationLayerInfo(descriptor);
125 
126  m_pDepthwiseConvolutionLayer = std::make_unique<arm_compute::NEDepthwiseConvolutionLayer>();
127  static_cast<arm_compute::NEDepthwiseConvolutionLayer*>(
128  m_pDepthwiseConvolutionLayer.get())->configure(&input,
129  m_KernelTensor.get(),
130  m_BiasTensor.get(),
131  &output,
132  padStrideInfo,
133  depthMultiplier,
134  activationInfo,
135  aclDilationInfo);
136 
137  ARMNN_ASSERT(m_pDepthwiseConvolutionLayer);
138 
139  ScopedCpuTensorHandle weightsPermutedHandle(weightPermuted);
140  InitializeArmComputeTensorData(*m_KernelTensor, &weightsPermutedHandle);
141 
142  if (m_Data.m_Parameters.m_BiasEnabled)
143  {
144  InitializeArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias);
145  }
146 
147  m_pDepthwiseConvolutionLayer->prepare();
148  FreeUnusedTensors();
149 }
armnn::IAclTensorHandle::GetTensor
virtual arm_compute::ITensor & GetTensor()=0
+
armnn::DepthwiseConvolution2dDescriptor::m_BiasEnabled
bool m_BiasEnabled
Enable/disable bias.
Definition: Descriptors.hpp:506
+
armnn::DataLayout
DataLayout
Definition: Types.hpp:50
+
armnn::ConvertWeightTensorFromArmnnToAcl
armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstCpuTensorHandle *weightTensor, DataLayout dataLayout, void *permuteBuffer)
Definition: WorkloadUtils.cpp:133
+
armnn::DepthwiseConvolution2dDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:508
+ +
armnn::BaseWorkload< DepthwiseConvolution2dQueueDescriptor >::m_Data
const DepthwiseConvolution2dQueueDescriptor m_Data
Definition: Workload.hpp:46
+
armnn::TensorInfo::GetNumBytes
unsigned int GetNumBytes() const
Definition: Tensor.cpp:418
+
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::DepthwiseConvolution2dDescriptor::m_DilationY
uint32_t m_DilationY
Dilation factor value for height dimension.
Definition: Descriptors.hpp:504
+ + +
armnn::DepthwiseConvolution2dDescriptor::m_DilationX
uint32_t m_DilationX
Dilation factor value for width dimension.
Definition: Descriptors.hpp:502
+
armnn::ConstTensor
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
Definition: Tensor.hpp:314
+
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::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 151 of file NeonDepthwiseConvolutionWorkload.cpp.

+ +

References ARMNN_ASSERT, and ARMNN_SCOPED_PROFILING_EVENT_NEON.

+
152 {
153  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonDepthwiseConvolutionWorkload_Execute");
154  ARMNN_ASSERT(m_pDepthwiseConvolutionLayer);
155 
156  m_pDepthwiseConvolutionLayer->run();
157 }
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+
ARMNN_ASSERT
#define ARMNN_ASSERT(COND)
Definition: Assert.hpp:14
+
+
+
+
The documentation for this class was generated from the following files: +
+
+ + + + -- cgit v1.2.1