1 //
2 // Copyright © 2022 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 {
RefArgMinMaxWorkload(const ArgMinMaxQueueDescriptor & descriptor,const WorkloadInfo & info)16 RefArgMinMaxWorkload::RefArgMinMaxWorkload(
17 const ArgMinMaxQueueDescriptor& descriptor,
18 const WorkloadInfo& info)
19 : RefBaseWorkload<ArgMinMaxQueueDescriptor>(descriptor, info) {}
20
21
Execute() const22 void RefArgMinMaxWorkload::Execute() const
23 {
24 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
25 }
26
ExecuteAsync(ExecutionData & executionData)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
Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const33 void RefArgMinMaxWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
34 {
35 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "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