From 6f92c8e9f8bb38dcf5dccf8deeff5112ecd8e37c Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Wed, 22 Nov 2023 11:41:15 +0000 Subject: Update Doxygen for 23.11 Signed-off-by: Nikhil Raj Change-Id: I47cd933f5002cb94a73aa97689d7b3d9c93cb849 --- 23.11/_ref_arg_min_max_workload_8cpp_source.html | 199 +++++++++++++++++++++++ 1 file changed, 199 insertions(+) create mode 100644 23.11/_ref_arg_min_max_workload_8cpp_source.html (limited to '23.11/_ref_arg_min_max_workload_8cpp_source.html') diff --git a/23.11/_ref_arg_min_max_workload_8cpp_source.html b/23.11/_ref_arg_min_max_workload_8cpp_source.html new file mode 100644 index 0000000000..0f8bfd217e --- /dev/null +++ b/23.11/_ref_arg_min_max_workload_8cpp_source.html @@ -0,0 +1,199 @@ + + + + + + + + +Arm NN: src/backends/reference/workloads/RefArgMinMaxWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  23.11 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
RefArgMinMaxWorkload.cpp
+
+
+Go to the documentation of this file.
1 //
+
2 // Copyright © 2019-2023 Arm Ltd and Contributors. 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  : RefBaseWorkload<ArgMinMaxQueueDescriptor>(descriptor, info) {}
+
20 
+
21 
+
22 void RefArgMinMaxWorkload::Execute() const
+
23 {
+
24  Execute(m_Data.m_Inputs, m_Data.m_Outputs);
+
25 }
+
26 
+
27 void RefArgMinMaxWorkload::ExecuteAsync(ExecutionData& executionData)
+
28 {
+
29  WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
+
30  Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs);
+
31 }
+
32 
+
33 void RefArgMinMaxWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
+
34 {
+
35  ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID("RefArgMinMaxWorkload_Execute");
+
36 
+
37  const TensorInfo &inputTensorInfo = GetTensorInfo(inputs[0]);
+
38 
+
39  std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputTensorInfo, inputs[0]->Map());
+
40  Decoder<float> &decoder = *decoderPtr;
+
41 
+
42  const TensorInfo &outputTensorInfo = GetTensorInfo(outputs[0]);
+
43 
+
44  if (outputTensorInfo.GetDataType() == armnn::DataType::Signed32) {
+
45  int32_t *output = GetOutputTensorData<int32_t>(outputs[0]);
+
46  ArgMinMax(decoder, output, inputTensorInfo, outputTensorInfo, m_Data.m_Parameters.m_Function,
+
47  m_Data.m_Parameters.m_Axis);
+
48  } else {
+
49  int64_t *output = GetOutputTensorData<int64_t>(outputs[0]);
+
50  ArgMinMax(decoder, output, inputTensorInfo, outputTensorInfo, m_Data.m_Parameters.m_Function,
+
51  m_Data.m_Parameters.m_Axis);
+
52  }
+
53 }
+
54 
+
55 } //namespace armnn
+
+
+
armnn::RefArgMinMaxWorkload::ExecuteAsync
void ExecuteAsync(ExecutionData &executionData) override
Definition: RefArgMinMaxWorkload.cpp:27
+ + + + +
armnn::ArgMinMaxDescriptor::m_Function
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Definition: Descriptors.hpp:81
+
ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID
#define ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID(label)
Creates a profiling event that uses GetGuid() and GetName() from the calling class.
Definition: RefWorkloadUtils.hpp:22
+
armnn::RefArgMinMaxWorkload::RefArgMinMaxWorkload
RefArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: RefArgMinMaxWorkload.cpp:16
+ +
armnn::ArgMinMax
void ArgMinMax(Decoder< float > &in, OUT *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
Definition: ArgMinMax.cpp:17
+
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
+
armnn::GetTensorInfo
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Definition: RefWorkloadUtils.hpp:33
+ + + +
armnn::TensorInfo::GetDataType
DataType GetDataType() const
Definition: Tensor.hpp:198
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:27
+ + + + + + + +
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
+
armnn::ArgMinMaxDescriptor::m_Axis
int m_Axis
Axis to reduce across the input tensor.
Definition: Descriptors.hpp:83
+ + + + + +
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:26
+ + + + + -- cgit v1.2.1