From 96becb7e4f5f510344c3850278a706d63a564fc4 Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Tue, 16 Jun 2020 12:41:49 +0100 Subject: Revert "IVGCVSW-3726 Upload ArmNN Doxygen files" This reverts commit de36e4a9c299028e792c3a5bd99ad0816d806077. Signed-off-by: Jan Eilers Change-Id: Idbf20c12ea07583ca552d7cc7fb517fbadc73fff --- .../_neon_arg_min_max_workload_8cpp_source.html | 125 --------------------- 1 file changed, 125 deletions(-) delete mode 100644 Documentation/_neon_arg_min_max_workload_8cpp_source.html (limited to 'Documentation/_neon_arg_min_max_workload_8cpp_source.html') diff --git a/Documentation/_neon_arg_min_max_workload_8cpp_source.html b/Documentation/_neon_arg_min_max_workload_8cpp_source.html deleted file mode 100644 index f27633dfa2..0000000000 --- a/Documentation/_neon_arg_min_max_workload_8cpp_source.html +++ /dev/null @@ -1,125 +0,0 @@ - - - - - - - -ArmNN: src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp Source File - - - - - - - - - - - - - - -
-
- - - - - - -
-
ArmNN -  NotReleased -
-
-
- - - - - - - -
-
- -
-
-
- -
- -
-
- - -
- -
- -
-
-
NeonArgMinMaxWorkload.cpp
-
-
-Go to the documentation of this file.
1 //
2 // Copyright © 2019 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "NeonArgMinMaxWorkload.hpp"
7 #include "NeonWorkloadUtils.hpp"
8 
9 #include <aclCommon/ArmComputeTensorUtils.hpp>
10 
11 #include <backendsCommon/CpuTensorHandle.hpp>
12 
13 #include <armnnUtils/TensorUtils.hpp>
14 
15 #include <arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h>
16 
17 namespace
18 {
19 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int axisIndex)
20 {
21  return (numDimensions - axisIndex) - 1;
22 }
23 
24 } //namespace
25 
26 namespace armnn
27 {
28 
29 arm_compute::Status NeonArgMinMaxWorkloadValidate(const TensorInfo& input,
30  const TensorInfo& output,
31  const ArgMinMaxDescriptor& descriptor)
32 {
33  const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
34  const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
35 
36  auto numDims = input.GetNumDimensions();
37  auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, descriptor.m_Axis);
38  int aclAxis = boost::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis));
39 
40  if (descriptor.m_Function == ArgMinMaxFunction::Max)
41  {
42  return arm_compute::NEArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
43  arm_compute::ReductionOperation::ARG_IDX_MAX);
44  }
45  else
46  {
47  return arm_compute::NEArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
48  arm_compute::ReductionOperation::ARG_IDX_MIN);
49  }
50 }
51 
52 
53 NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(const ArgMinMaxQueueDescriptor& descriptor,
54  const WorkloadInfo& info)
55  : BaseWorkload<ArgMinMaxQueueDescriptor>(descriptor, info)
56 {
57  arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
58  arm_compute::ITensor& output = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
59 
60  auto numDims = info.m_InputTensorInfos[0].GetNumDimensions();
61  auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, m_Data.m_Parameters.m_Axis);
62  int aclAxis = boost::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis));
63 
64  auto layer = std::make_unique<arm_compute::NEArgMinMaxLayer>();
65 
66  if (m_Data.m_Parameters.m_Function == ArgMinMaxFunction::Max)
67  {
68  layer->configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MAX);
69  }
70  else
71  {
72  layer->configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MIN);
73  }
74 
75  m_ArgMinMaxLayer.reset(layer.release());
76 }
77 
78 void NeonArgMinMaxWorkload::Execute() const
79 {
80  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonArgMinMaxWorkload_Execute");
81  m_ArgMinMaxLayer->run();
82 }
83 
84 } //namespace armnn
85 
-
armnn::NeonArgMinMaxWorkloadValidate
arm_compute::Status NeonArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
Definition: NeonArgMinMaxWorkload.cpp:29
-
armnn::Status
Status
Definition: Types.hpp:26
-
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
-
armnnUtils::GetUnsignedAxis
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
Definition: TensorUtils.cpp:127
-
armnn::TensorInfo::GetNumDimensions
unsigned int GetNumDimensions() const
Definition: Tensor.hpp:92
- - -
TensorUtils.hpp
-
NeonWorkloadUtils.hpp
- -
armnn::ArgMinMaxDescriptor::m_Function
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Definition: Descriptors.hpp:56
-
armnn::ArgMinMaxDescriptor::m_Axis
int m_Axis
Axis to reduce across the input tensor.
Definition: Descriptors.hpp:58
-
ArmComputeTensorUtils.hpp
-
armnn::BaseWorkload< ArgMinMaxQueueDescriptor >::m_Data
const ArgMinMaxQueueDescriptor m_Data
Definition: Workload.hpp:46
-
armnn::WorkloadInfo::m_InputTensorInfos
std::vector< TensorInfo > m_InputTensorInfos
Definition: WorkloadInfo.hpp:18
- -
CpuTensorHandle.hpp
-
NeonArgMinMaxWorkload.hpp
-
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::NeonArgMinMaxWorkload::NeonArgMinMaxWorkload
NeonArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonArgMinMaxWorkload.cpp:53
-
armnn::NeonArgMinMaxWorkload::Execute
virtual void Execute() const override
Definition: NeonArgMinMaxWorkload.cpp:78
-
armnn::ArgMinMaxDescriptor
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:43
- - -
-
- - - - -- cgit v1.2.1