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/_cl_mean_workload_8cpp_source.xhtml | 140 ++++++++++++++++++++++++++++++ 1 file changed, 140 insertions(+) create mode 100644 21.02/_cl_mean_workload_8cpp_source.xhtml (limited to '21.02/_cl_mean_workload_8cpp_source.xhtml') diff --git a/21.02/_cl_mean_workload_8cpp_source.xhtml b/21.02/_cl_mean_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..2f0441b45b --- /dev/null +++ b/21.02/_cl_mean_workload_8cpp_source.xhtml @@ -0,0 +1,140 @@ + + + + + + + + + + + + + +ArmNN: src/backends/cl/workloads/ClMeanWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
ClMeanWorkload.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 "ClMeanWorkload.hpp"
7 
8 #include <cl/ClTensorHandle.hpp>
10 
11 #include "ClWorkloadUtils.hpp"
12 
13 namespace armnn
14 {
15 using namespace armcomputetensorutils;
16 
18  const TensorInfo& output,
19  const MeanDescriptor& desc)
20 {
21  const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(input);
22  const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output);
23 
24  arm_compute::Coordinates coords = BuildArmComputeReductionCoordinates(aclInputInfo.num_dimensions(),
25  input.GetNumDimensions(),
26  desc.m_Axis);
27 
28  return arm_compute::CLReduceMean::validate(&aclInputInfo, coords, desc.m_KeepDims, &aclOutputInfo);
29 }
30 
32  const WorkloadInfo& info,
33  const arm_compute::CLCompileContext& clCompileContext)
34  : BaseWorkload<MeanQueueDescriptor>(descriptor, info)
35 {
36  m_Data.ValidateInputsOutputs("ClMeanWorkload", 1, 1);
37 
38  arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
39  arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
40 
41  arm_compute::Coordinates coords = BuildArmComputeReductionCoordinates(input.info()->num_dimensions(),
42  info.m_InputTensorInfos[0].GetNumDimensions(),
44 
45  m_Layer.configure(clCompileContext, &input, coords, m_Data.m_Parameters.m_KeepDims, &output);
46 }
47 
49 {
50  ARMNN_SCOPED_PROFILING_EVENT_CL("ClMeanWorkload_Execute");
51  m_Layer.run();
52 }
53 
54 } //namespace armnn
+
void Execute() const override
+
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)
+
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
+
const MeanQueueDescriptor m_Data
Definition: Workload.hpp:46
+ +
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
+
Copyright (c) 2021 ARM Limited and Contributors.
+ +
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
+
std::vector< TensorInfo > m_InputTensorInfos
+ +
bool m_KeepDims
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept...
+ +
Status
enumeration
Definition: Types.hpp:26
+
arm_compute::Status ClMeanValidate(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &desc)
+ +
std::vector< ITensorHandle * > m_Outputs
+ + +
A MeanDescriptor for the MeanLayer.
+
Contains information about inputs and outputs to a layer.
+
std::vector< ITensorHandle * > m_Inputs
+ +
ClMeanWorkload(const MeanQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
+ +
unsigned int GetNumDimensions() const
Definition: Tensor.hpp:191
+
+
+ + + + -- cgit v1.2.1