From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/_ref_arg_min_max_workload_8cpp_source.xhtml | 137 ++++++++++++++++++++++ 1 file changed, 137 insertions(+) create mode 100644 20.02/_ref_arg_min_max_workload_8cpp_source.xhtml (limited to '20.02/_ref_arg_min_max_workload_8cpp_source.xhtml') diff --git a/20.02/_ref_arg_min_max_workload_8cpp_source.xhtml b/20.02/_ref_arg_min_max_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..1cbb534bf5 --- /dev/null +++ b/20.02/_ref_arg_min_max_workload_8cpp_source.xhtml @@ -0,0 +1,137 @@ + + + + + + + + + + + + + +ArmNN: src/backends/reference/workloads/RefArgMinMaxWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
RefArgMinMaxWorkload.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 "RefArgMinMaxWorkload.hpp"
7 
8 #include "ArgMinMax.hpp"
9 #include "RefWorkloadUtils.hpp"
10 #include "Decoders.hpp"
11 #include "Encoders.hpp"
12 #include "Profiling.hpp"
13 
14 namespace armnn
15 {
16 RefArgMinMaxWorkload::RefArgMinMaxWorkload(
17  const ArgMinMaxQueueDescriptor& descriptor,
18  const WorkloadInfo& info)
19  : BaseWorkload<ArgMinMaxQueueDescriptor>(descriptor, info) {}
20 
21 void RefArgMinMaxWorkload::Execute() const
22 {
23  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefArgMinMaxWorkload_Execute");
24 
25  const TensorInfo &inputTensorInfo = GetTensorInfo(m_Data.m_Inputs[0]);
26 
27  std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputTensorInfo, m_Data.m_Inputs[0]->Map());
28  Decoder<float> &decoder = *decoderPtr;
29 
30  const TensorInfo &outputTensorInfo = GetTensorInfo(m_Data.m_Outputs[0]);
31 
32  int32_t* output = GetOutputTensorData<int32_t>(0, m_Data);
33 
34  ArgMinMax(decoder, output, inputTensorInfo, outputTensorInfo, m_Data.m_Parameters.m_Function,
35  m_Data.m_Parameters.m_Axis);
36 }
37 
38 } //namespace armnn
armnn::Compute::CpuRef
CPU Execution: Reference C++ kernels.
+
RefWorkloadUtils.hpp
+ +
armnn::BaseWorkload< ArgMinMaxQueueDescriptor >::m_Data
const ArgMinMaxQueueDescriptor m_Data
Definition: Workload.hpp:46
+
armnn::ArgMinMaxDescriptor::m_Function
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Definition: Descriptors.hpp:56
+
armnn::GetTensorInfo
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Definition: RefWorkloadUtils.hpp:25
+
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
+ +
ARMNN_SCOPED_PROFILING_EVENT
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:169
+
Encoders.hpp
+ +
armnn::RefArgMinMaxWorkload::Execute
virtual void Execute() const override
Definition: RefArgMinMaxWorkload.cpp:21
+
Decoders.hpp
+
RefArgMinMaxWorkload.hpp
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+ +
armnn::RefArgMinMaxWorkload::RefArgMinMaxWorkload
RefArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: RefArgMinMaxWorkload.cpp:16
+ +
ArgMinMax.hpp
+
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
+
armnn::ArgMinMaxDescriptor::m_Axis
int m_Axis
Axis to reduce across the input tensor.
Definition: Descriptors.hpp:58
+
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
Profiling.hpp
+ +
+
+ + + + -- cgit v1.2.1