From 8efb48a6847c5cd166c561127ae6611150963ce3 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Fri, 19 May 2023 11:14:28 +0100 Subject: Update Doxygen docu for 23.05 Signed-off-by: Nikhil Raj Change-Id: I0a992286f14fa68fcc6e5eba31ac39fed003cbbe --- 23.05/_neon_mean_workload_8cpp_source.xhtml | 203 ++++++++++++++++++++++++++++ 1 file changed, 203 insertions(+) create mode 100644 23.05/_neon_mean_workload_8cpp_source.xhtml (limited to '23.05/_neon_mean_workload_8cpp_source.xhtml') diff --git a/23.05/_neon_mean_workload_8cpp_source.xhtml b/23.05/_neon_mean_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..07b17e102d --- /dev/null +++ b/23.05/_neon_mean_workload_8cpp_source.xhtml @@ -0,0 +1,203 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonMeanWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  23.05 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonMeanWorkload.cpp
+
+
+Go to the documentation of this file.
1 //
+
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
+
3 // SPDX-License-Identifier: MIT
+
4 //
+
5 
+
6 #include "NeonMeanWorkload.hpp"
+
7 
+ +
9 
+ +
11 
+
12 #include "NeonWorkloadUtils.hpp"
+
13 
+
14 namespace armnn
+
15 {
+
16 using namespace armcomputetensorutils;
+
17 
+ +
19  const TensorInfo& output,
+
20  const MeanDescriptor& descriptor)
+
21 {
+
22  const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(input);
+
23  const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output);
+
24 
+
25  arm_compute::Coordinates coords = BuildArmComputeReductionCoordinates(aclInputInfo.num_dimensions(),
+
26  input.GetNumDimensions(),
+
27  descriptor.m_Axis);
+
28 
+
29  return arm_compute::NEReduceMean::validate(&aclInputInfo, coords, descriptor.m_KeepDims, &aclOutputInfo);
+
30 }
+
31 
+ + +
34 {
+
35  // Report Profiling Details
+
36  ARMNN_REPORT_PROFILING_WORKLOAD_DESC("NeonMeanWorkload_Construct",
+
37  descriptor.m_Parameters,
+
38  info,
+
39  this->GetGuid());
+
40 
+
41  m_Data.ValidateInputsOutputs("NeonMeanWorkload", 1, 1);
+
42 
+
43  arm_compute::ITensor& input = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
+
44  arm_compute::ITensor& output = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
+
45 
+
46  arm_compute::Coordinates coords = BuildArmComputeReductionCoordinates(input.info()->num_dimensions(),
+
47  info.m_InputTensorInfos[0].GetNumDimensions(),
+ +
49 
+
50  m_Layer.configure(&input, coords, m_Data.m_Parameters.m_KeepDims, &output);
+
51 }
+
52 
+ +
54 {
+
55  ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID("NeonMeanWorkload_Execute", this->GetGuid());
+
56  m_Layer.run();
+
57 }
+
58 
+
59 } //namespace armnn
+
+
+
arm::pipe::ProfilingGuid GetGuid() const final
Definition: Workload.hpp:61
+
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
+ +
void Execute() const override
+
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
+ +
NeonMeanWorkload(const MeanQueueDescriptor &descriptor, const WorkloadInfo &info)
+
Copyright (c) 2021 ARM Limited and Contributors.
+
#define ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID(name, guid)
+
arm_compute::Status NeonMeanWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &descriptor)
+
A MeanDescriptor for the MeanLayer.
+
unsigned int GetNumDimensions() const
Definition: Tensor.hpp:195
+ + + + +
Status
Definition: Types.hpp:42
+
Contains information about TensorInfos of a layer.
+ +
#define ARMNN_REPORT_PROFILING_WORKLOAD_DESC(name, desc, infos, guid)
Definition: Profiling.hpp:227
+ +
bool m_KeepDims
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.
+
std::vector< ITensorHandle * > m_Outputs
+ + +
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
+
std::vector< ITensorHandle * > m_Inputs
+ + + + + -- cgit v1.2.1