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 --- 21.02/_neon_concat_workload_8cpp_source.xhtml | 138 ++++++++++++++++++++++++++ 1 file changed, 138 insertions(+) create mode 100644 21.02/_neon_concat_workload_8cpp_source.xhtml (limited to '21.02/_neon_concat_workload_8cpp_source.xhtml') diff --git a/21.02/_neon_concat_workload_8cpp_source.xhtml b/21.02/_neon_concat_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..5604568d9c --- /dev/null +++ b/21.02/_neon_concat_workload_8cpp_source.xhtml @@ -0,0 +1,138 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonConcatWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonConcatWorkload.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "NeonConcatWorkload.hpp"
7 
8 #include "NeonWorkloadUtils.hpp"
9 
10 #include <aclCommon/ArmComputeTensorUtils.hpp>
11 #include <armnn/utility/PolymorphicDowncast.hpp>
12 #include <backendsCommon/CpuTensorHandle.hpp>
13 #include <neon/NeonTensorHandle.hpp>
14 
15 namespace armnn
16 {
17 using namespace armcomputetensorutils;
18 
19 namespace
20 {
21 size_t CalcAxis(const armnn::OriginsDescriptor& desc)
22 {
23  return (desc.GetNumDimensions() - desc.GetConcatAxis()) - 1;
24 }
25 } //namespace
26 
27 arm_compute::Status NeonConcatWorkloadValidate(const std::vector<const TensorInfo*>& inputs,
28  const TensorInfo& output,
29  const OriginsDescriptor& descriptor)
30 
31 {
32  std::vector<arm_compute::TensorInfo> aclInputs;
33  for (const TensorInfo* input : inputs)
34  {
35  arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(*input, armnn::DataLayout::NCHW);
36  aclInputs.emplace_back(aclInputInfo);
37  }
38  const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
39  std::vector<const arm_compute::ITensorInfo*> aclInputPtrs;
40  for (arm_compute::ITensorInfo& input : aclInputs)
41  {
42  aclInputPtrs.emplace_back(&input);
43  }
44 
45  size_t aclAxis = CalcAxis(descriptor);
46  return arm_compute::NEConcatenateLayer::validate(aclInputPtrs, &aclOutputInfo, aclAxis);
47 }
48 
49 NeonConcatWorkload::NeonConcatWorkload(
50 const ConcatQueueDescriptor& descriptor, const WorkloadInfo& info)
51  : BaseWorkload<ConcatQueueDescriptor>(descriptor, info)
52 {
53  bool allInputsAreSubtensors = true;
54 
55  // Check that all inputs are sub-tensors
56  for (auto input : descriptor.m_Inputs)
57  {
58  if (!input->GetParent())
59  {
60  // Non sub-tensor input found so we need to execute the concat function
61  allInputsAreSubtensors = false;
62  break;
63  }
64  }
65 
66  if (allInputsAreSubtensors)
67  {
68  // Can skip configuring the concat function since it's not executed
69  return;
70  }
71 
72  std::vector<const arm_compute::ITensor *> aclInputs;
73  for (auto input : m_Data.m_Inputs)
74  {
75  arm_compute::ITensor& aclInput = armnn::PolymorphicPointerDowncast<IAclTensorHandle>(input)->GetTensor();
76  aclInputs.emplace_back(&aclInput);
77  }
78  arm_compute::ITensor& output = armnn::PolymorphicPointerDowncast<IAclTensorHandle>(
79  m_Data.m_Outputs[0])->GetTensor();
80 
81  // Create the layer function
82  m_Layer.reset(new arm_compute::NEConcatenateLayer());
83 
84  // Configure input and output tensors
85  size_t aclAxis = CalcAxis(descriptor.m_Parameters);
86  m_Layer->configure(aclInputs, &output, aclAxis);
87 
88  // Prepare
89  m_Layer->prepare();
90 }
91 
92 void NeonConcatWorkload::Execute() const
93 {
94  if (m_Layer)
95  {
96  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonConcatWorkload_Execute");
97  m_Layer->run();
98  }
99 }
100 
101 } //namespace armnn
102 
+
NeonConcatWorkload.hpp
+
armnn::BaseWorkload< ConcatQueueDescriptor >::m_Data
const ConcatQueueDescriptor m_Data
Definition: Workload.hpp:46
+
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+
armnn::NeonConcatWorkload::NeonConcatWorkload
NeonConcatWorkload(const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonConcatWorkload.cpp:49
+
NeonTensorHandle.hpp
+
ArmComputeTensorUtils.hpp
+
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_software_tools.dox:6
+ + +
PolymorphicDowncast.hpp
+ +
armnn::OriginsDescriptor
An OriginsDescriptor for the ConcatLayer.
Definition: Descriptors.hpp:163
+
armnn::NeonConcatWorkload::Execute
void Execute() const override
Definition: NeonConcatWorkload.cpp:92
+
armnn::Status
Status
enumeration
Definition: Types.hpp:26
+
armnn::NeonConcatWorkloadValidate
arm_compute::Status NeonConcatWorkloadValidate(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const OriginsDescriptor &descriptor)
Definition: NeonConcatWorkload.cpp:27
+
NeonWorkloadUtils.hpp
+
CpuTensorHandle.hpp
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+ +
armnn::OriginsDescriptor::GetNumDimensions
uint32_t GetNumDimensions() const
Get the number of dimensions.
Definition: Descriptors.cpp:192
+
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
+
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+ +
armnn::OriginsDescriptor::GetConcatAxis
unsigned int GetConcatAxis() const
Get the concatenation axis value.
Definition: Descriptors.cpp:162
+
+
+ + + + -- cgit v1.2.1