ArmNN
 21.11
ArgMinMaxTestImpl.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "ArgMinMaxTestImpl.hpp"
7 
8 
9 #include <backendsCommon/test/DataTypeUtils.hpp>
10 #include <backendsCommon/test/TensorCopyUtils.hpp>
11 #include <backendsCommon/test/WorkloadTestUtils.hpp>
12 
13 #include <test/TensorHelpers.hpp>
14 
15 namespace
16 {
17 
18 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
19 LayerTestResult<int32_t, 3> ArgMinMaxTestCommon(
20  armnn::IWorkloadFactory& workloadFactory,
21  const armnn::IBackendInternal::IMemoryManagerSharedPtr&,
22  const armnn::ITensorHandleFactory& tensorHandleFactory,
23  armnn::ArgMinMaxFunction argMinMaxFunction,
24  const armnn::TensorInfo inputTensorInfo,
25  const armnn::TensorInfo outputTensorInfo,
26  const std::vector<float>& inputData,
27  const std::vector<int32_t>& outputData,
28  int axis = 3)
29 {
30  std::vector<T> inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo);
31  std::vector<int32_t> actualOutput(outputTensorInfo.GetNumElements());
32 
33  std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
34  std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
35 
36  armnn::ArgMinMaxQueueDescriptor descriptor;
37  descriptor.m_Parameters.m_Function = argMinMaxFunction;
38  descriptor.m_Parameters.m_Axis = axis;
39  armnn::WorkloadInfo info;
40 
41  AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get());
42  AddOutputToWorkload(descriptor, info, outputTensorInfo, outputHandle.get());
43 
44  std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateArgMinMax(descriptor, info);
45 
46  inputHandle->Allocate();
47  outputHandle->Allocate();
48 
49  CopyDataToITensorHandle(inputHandle.get(), inputTensor.data());
50 
51  workload->PostAllocationConfigure();
52  workload->Execute();
53 
54  CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get());
55 
56  return LayerTestResult<int32_t, 3>(actualOutput,
57  outputData,
58  outputHandle->GetShape(),
59  outputTensorInfo.GetShape());
60 }
61 
62 } // namespace
63 
64 template<armnn::DataType ArmnnType, typename T>
65 LayerTestResult<int32_t, 3> ArgMaxSimpleTest(
66  armnn::IWorkloadFactory& workloadFactory,
67  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
68  const armnn::ITensorHandleFactory& tensorHandleFactory)
69 {
70  const armnn::TensorShape inputShape{ 1, 1, 1, 5 };
71  const armnn::TensorShape outputShape{ 1, 1, 1 };
72 
73  armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType);
74 
75  if (armnn::IsQuantizedType<T>())
76  {
77  inputTensorInfo.SetQuantizationScale(1.0f);
78  inputTensorInfo.SetQuantizationOffset(0);
79  }
80 
81  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
82 
83  std::vector<float> inputValues({ 5.0f, 2.0f, 8.0f, 10.0f, 9.0f });
84  std::vector<int32_t> outputValues({ 3 });
85 
86  return ArgMinMaxTestCommon<ArmnnType>(workloadFactory, memoryManager, tensorHandleFactory,
87  armnn::ArgMinMaxFunction::Max,
88  inputTensorInfo, outputTensorInfo,
89  inputValues, outputValues, -1); // axis -1 === 3
90 }
91 
92 template<armnn::DataType ArmnnType, typename T>
93 LayerTestResult<int32_t, 3> ArgMinSimpleTest(
94  armnn::IWorkloadFactory& workloadFactory,
95  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
96  const armnn::ITensorHandleFactory& tensorHandleFactory)
97 {
98  const armnn::TensorShape inputShape{ 1, 1, 1, 5 };
99  const armnn::TensorShape outputShape{ 1, 1, 1 };
100 
101  armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType);
102 
103  if (armnn::IsQuantizedType<T>())
104  {
105  inputTensorInfo.SetQuantizationScale(1.0f);
106  inputTensorInfo.SetQuantizationOffset(0);
107  }
108 
109  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
110 
111  std::vector<float> inputValues({ 5.0f, 2.0f, 8.0f, 10.0f, 9.0f });
112  std::vector<int32_t> outputValues({ 1 });
113 
114  return ArgMinMaxTestCommon<ArmnnType>(workloadFactory, memoryManager, tensorHandleFactory,
115  armnn::ArgMinMaxFunction::Min,
116  inputTensorInfo, outputTensorInfo,
117  inputValues, outputValues, 3);
118 }
119 
120 template<armnn::DataType ArmnnType, typename T>
121 LayerTestResult<int32_t, 3> ArgMinChannelTest(
122  armnn::IWorkloadFactory& workloadFactory,
123  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
124  const armnn::ITensorHandleFactory& tensorHandleFactory)
125 {
126  const armnn::TensorShape inputShape{ 1, 3, 2, 4};
127  const armnn::TensorShape outputShape{ 1, 2, 4 };
128 
129  armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType);
130 
131  if (armnn::IsQuantizedType<T>())
132  {
133  inputTensorInfo.SetQuantizationScale(1.0f);
134  inputTensorInfo.SetQuantizationOffset(0);
135  }
136 
137  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
138 
139  std::vector<float> inputValues({ 1.0f, 2.0f, 3.0f, 4.0f,
140  5.0f, 6.0f, 7.0f, 8.0f,
141 
142  10.0f, 20.0f, 30.0f, 40.0f,
143  50.0f, 60.0f, 70.0f, 80.0f,
144 
145  100.0f, 200.0f, 300.0f, 400.0f,
146  500.0f, 600.0f, 700.0f, 800.0f });
147  std::vector<int32_t> outputValues({ 0, 0, 0, 0,
148  0, 0, 0, 0 });
149 
150  return ArgMinMaxTestCommon<ArmnnType>(workloadFactory, memoryManager, tensorHandleFactory,
151  armnn::ArgMinMaxFunction::Min,
152  inputTensorInfo, outputTensorInfo,
153  inputValues, outputValues, 1);
154 }
155 
156 template<armnn::DataType ArmnnType, typename T>
157 LayerTestResult<int32_t, 3> ArgMaxChannelTest(
158  armnn::IWorkloadFactory& workloadFactory,
159  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
160  const armnn::ITensorHandleFactory& tensorHandleFactory)
161 {
162  const armnn::TensorShape inputShape{ 1, 3, 2, 4};
163  const armnn::TensorShape outputShape{ 1, 2, 4 };
164 
165  armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType);
166 
167  if (armnn::IsQuantizedType<T>())
168  {
169  inputTensorInfo.SetQuantizationScale(1.0f);
170  inputTensorInfo.SetQuantizationOffset(0);
171  }
172 
173  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
174 
175  std::vector<float> inputValues({ 1.0f, 2.0f, 3.0f, 4.0f,
176  5.0f, 6.0f, 7.0f, 8.0f,
177 
178  10.0f, 20.0f, 30.0f, 40.0f,
179  50.0f, 60.0f, 70.0f, 80.0f,
180 
181  100.0f, 200.0f, 300.0f, 400.0f,
182  500.0f, 600.0f, 700.0f, 800.0f });
183  std::vector<int32_t> outputValues({ 2, 2, 2, 2,
184  2, 2, 2, 2 });
185 
186  return ArgMinMaxTestCommon<ArmnnType>(workloadFactory, memoryManager, tensorHandleFactory,
187  armnn::ArgMinMaxFunction::Max,
188  inputTensorInfo, outputTensorInfo,
189  inputValues, outputValues, 1);
190 }
191 
192 template<armnn::DataType ArmnnType, typename T>
193 LayerTestResult<int32_t, 3> ArgMaxHeightTest(
194  armnn::IWorkloadFactory& workloadFactory,
195  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
196  const armnn::ITensorHandleFactory& tensorHandleFactory)
197 {
198  const armnn::TensorShape inputShape{ 1, 3, 2, 4};
199  const armnn::TensorShape outputShape{ 1, 3, 4 };
200 
201  armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType);
202  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
203 
204  if (armnn::IsQuantizedType<T>())
205  {
206  inputTensorInfo.SetQuantizationScale(1.0f);
207  inputTensorInfo.SetQuantizationOffset(0);
208  }
209 
210  std::vector<float> inputValues({ 1.0f, 2.0f, 3.0f, 4.0f,
211  5.0f, 6.0f, 7.0f, 8.0f,
212 
213  10.0f, 20.0f, 30.0f, 40.0f,
214  50.0f, 60.0f, 70.0f, 80.0f,
215 
216  100.0f, 200.0f, 300.0f, 400.0f,
217  500.0f, 600.0f, 700.0f, 800.0f });
218  std::vector<int32_t> outputValues({ 1, 1, 1, 1,
219  1, 1, 1, 1,
220  1, 1, 1, 1 });
221 
222  return ArgMinMaxTestCommon<ArmnnType>(workloadFactory, memoryManager, tensorHandleFactory,
223  armnn::ArgMinMaxFunction::Max,
224  inputTensorInfo, outputTensorInfo,
225  inputValues, outputValues, 2);
226 }
227 
228 template<armnn::DataType ArmnnType, typename T>
229 LayerTestResult<int32_t, 3> ArgMinWidthTest(
230  armnn::IWorkloadFactory& workloadFactory,
231  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
232  const armnn::ITensorHandleFactory& tensorHandleFactory)
233 {
234  const armnn::TensorShape inputShape{ 1, 3, 2, 4};
235  const armnn::TensorShape outputShape{ 1, 3, 2 };
236 
237  armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType);
238  armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32);
239 
240  if (armnn::IsQuantizedType<T>())
241  {
242  inputTensorInfo.SetQuantizationScale(1.0f);
243  inputTensorInfo.SetQuantizationOffset(0);
244  }
245 
246  std::vector<float> inputValues({ 1.0f, 2.0f, 3.0f, 4.0f,
247  5.0f, 6.0f, 7.0f, 8.0f,
248 
249  10.0f, 20.0f, 30.0f, 40.0f,
250  50.0f, 60.0f, 70.0f, 80.0f,
251 
252  100.0f, 200.0f, 300.0f, 400.0f,
253  500.0f, 600.0f, 700.0f, 800.0f });
254  std::vector<int32_t> outputValues({ 0, 0,
255  0, 0,
256  0, 0 });
257 
258  return ArgMinMaxTestCommon<ArmnnType>(workloadFactory, memoryManager, tensorHandleFactory,
259  armnn::ArgMinMaxFunction::Min,
260  inputTensorInfo, outputTensorInfo,
261  inputValues, outputValues, 3);
262 }
263 
264 
265 // Explicit template specializations
266 
267 template LayerTestResult<int32_t, 3>
268 ArgMaxSimpleTest<armnn::DataType::Float32>(
269  armnn::IWorkloadFactory& workloadFactory,
270  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
271  const armnn::ITensorHandleFactory& tensorHandleFactory);
272 
273 template LayerTestResult<int32_t, 3>
274 ArgMaxSimpleTest<armnn::DataType::Float16>(
275  armnn::IWorkloadFactory& workloadFactory,
276  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
277  const armnn::ITensorHandleFactory& tensorHandleFactory);
278 
279 template LayerTestResult<int32_t, 3>
280 ArgMaxSimpleTest<armnn::DataType::QAsymmS8>(
281  armnn::IWorkloadFactory& workloadFactory,
282  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
283  const armnn::ITensorHandleFactory& tensorHandleFactory);
284 
285 template LayerTestResult<int32_t, 3>
286 ArgMaxSimpleTest<armnn::DataType::QAsymmU8>(
287  armnn::IWorkloadFactory& workloadFactory,
288  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
289  const armnn::ITensorHandleFactory& tensorHandleFactory);
290 
291 template LayerTestResult<int32_t, 3>
292 ArgMaxSimpleTest<armnn::DataType::QSymmS16>(
293  armnn::IWorkloadFactory& workloadFactory,
294  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
295  const armnn::ITensorHandleFactory& tensorHandleFactory);
296 
297 template LayerTestResult<int32_t, 3>
298 ArgMaxSimpleTest<armnn::DataType::Signed32>(
299  armnn::IWorkloadFactory& workloadFactory,
300  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
301  const armnn::ITensorHandleFactory& tensorHandleFactory);
302 
303 template LayerTestResult<int32_t, 3>
304 ArgMinSimpleTest<armnn::DataType::Float32>(
305  armnn::IWorkloadFactory& workloadFactory,
306  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
307  const armnn::ITensorHandleFactory& tensorHandleFactory);
308 
309 template LayerTestResult<int32_t, 3>
310 ArgMinSimpleTest<armnn::DataType::Float16>(
311  armnn::IWorkloadFactory& workloadFactory,
312  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
313  const armnn::ITensorHandleFactory& tensorHandleFactory);
314 
315 template LayerTestResult<int32_t, 3>
316 ArgMinSimpleTest<armnn::DataType::QAsymmS8>(
317  armnn::IWorkloadFactory& workloadFactory,
318  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
319  const armnn::ITensorHandleFactory& tensorHandleFactory);
320 
321 template LayerTestResult<int32_t, 3>
322 ArgMinSimpleTest<armnn::DataType::QAsymmU8>(
323  armnn::IWorkloadFactory& workloadFactory,
324  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
325  const armnn::ITensorHandleFactory& tensorHandleFactory);
326 
327 template LayerTestResult<int32_t, 3>
328 ArgMinSimpleTest<armnn::DataType::QSymmS16>(
329  armnn::IWorkloadFactory& workloadFactory,
330  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
331  const armnn::ITensorHandleFactory& tensorHandleFactory);
332 
333 template LayerTestResult<int32_t, 3>
334 ArgMinSimpleTest<armnn::DataType::Signed32>(
335  armnn::IWorkloadFactory& workloadFactory,
336  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
337  const armnn::ITensorHandleFactory& tensorHandleFactory);
338 
339 template LayerTestResult<int32_t, 3>
340 ArgMinChannelTest<armnn::DataType::Float32>(
341  armnn::IWorkloadFactory& workloadFactory,
342  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
343  const armnn::ITensorHandleFactory& tensorHandleFactory);
344 
345 template LayerTestResult<int32_t, 3>
346 ArgMinChannelTest<armnn::DataType::Float16>(
347  armnn::IWorkloadFactory& workloadFactory,
348  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
349  const armnn::ITensorHandleFactory& tensorHandleFactory);
350 
351 template LayerTestResult<int32_t, 3>
352 ArgMinChannelTest<armnn::DataType::QAsymmS8>(
353  armnn::IWorkloadFactory& workloadFactory,
354  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
355  const armnn::ITensorHandleFactory& tensorHandleFactory);
356 
357 template LayerTestResult<int32_t, 3>
358 ArgMinChannelTest<armnn::DataType::QAsymmU8>(
359  armnn::IWorkloadFactory& workloadFactory,
360  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
361  const armnn::ITensorHandleFactory& tensorHandleFactory);
362 
363 template LayerTestResult<int32_t, 3>
364 ArgMinChannelTest<armnn::DataType::QSymmS16>(
365  armnn::IWorkloadFactory& workloadFactory,
366  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
367  const armnn::ITensorHandleFactory& tensorHandleFactory);
368 
369 template LayerTestResult<int32_t, 3>
370 ArgMinChannelTest<armnn::DataType::Signed32>(
371  armnn::IWorkloadFactory& workloadFactory,
372  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
373  const armnn::ITensorHandleFactory& tensorHandleFactory);
374 
375 template LayerTestResult<int32_t, 3>
376 ArgMaxChannelTest<armnn::DataType::Float32>(
377  armnn::IWorkloadFactory& workloadFactory,
378  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
379  const armnn::ITensorHandleFactory& tensorHandleFactory);
380 
381 template LayerTestResult<int32_t, 3>
382 ArgMaxChannelTest<armnn::DataType::Float16>(
383  armnn::IWorkloadFactory& workloadFactory,
384  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
385  const armnn::ITensorHandleFactory& tensorHandleFactory);
386 
387 template LayerTestResult<int32_t, 3>
388 ArgMaxChannelTest<armnn::DataType::QAsymmS8>(
389  armnn::IWorkloadFactory& workloadFactory,
390  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
391  const armnn::ITensorHandleFactory& tensorHandleFactory);
392 
393 template LayerTestResult<int32_t, 3>
394 ArgMaxChannelTest<armnn::DataType::QAsymmU8>(
395  armnn::IWorkloadFactory& workloadFactory,
396  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
397  const armnn::ITensorHandleFactory& tensorHandleFactory);
398 
399 template LayerTestResult<int32_t, 3>
400 ArgMaxChannelTest<armnn::DataType::QSymmS16>(
401  armnn::IWorkloadFactory& workloadFactory,
402  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
403  const armnn::ITensorHandleFactory& tensorHandleFactory);
404 
405 template LayerTestResult<int32_t, 3>
406 ArgMaxChannelTest<armnn::DataType::Signed32>(
407  armnn::IWorkloadFactory& workloadFactory,
408  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
409  const armnn::ITensorHandleFactory& tensorHandleFactory);
410 
411 template LayerTestResult<int32_t, 3>
412 ArgMaxHeightTest<armnn::DataType::Float32>(
413  armnn::IWorkloadFactory& workloadFactory,
414  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
415  const armnn::ITensorHandleFactory& tensorHandleFactory);
416 
417 template LayerTestResult<int32_t, 3>
418 ArgMaxHeightTest<armnn::DataType::Float16>(
419  armnn::IWorkloadFactory& workloadFactory,
420  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
421  const armnn::ITensorHandleFactory& tensorHandleFactory);
422 
423 template LayerTestResult<int32_t, 3>
424 ArgMaxHeightTest<armnn::DataType::Signed32>(
425  armnn::IWorkloadFactory& workloadFactory,
426  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
427  const armnn::ITensorHandleFactory& tensorHandleFactory);
428 
429 template LayerTestResult<int32_t, 3>
430 ArgMaxHeightTest<armnn::DataType::QAsymmS8>(
431  armnn::IWorkloadFactory& workloadFactory,
432  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
433  const armnn::ITensorHandleFactory& tensorHandleFactory);
434 
435 template LayerTestResult<int32_t, 3>
436 ArgMaxHeightTest<armnn::DataType::QAsymmU8>(
437  armnn::IWorkloadFactory& workloadFactory,
438  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
439  const armnn::ITensorHandleFactory& tensorHandleFactory);
440 
441 template LayerTestResult<int32_t, 3>
442 ArgMinWidthTest<armnn::DataType::Float32>(
443  armnn::IWorkloadFactory& workloadFactory,
444  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
445  const armnn::ITensorHandleFactory& tensorHandleFactory);
446 
447 template LayerTestResult<int32_t, 3>
448 ArgMinWidthTest<armnn::DataType::Float16>(
449  armnn::IWorkloadFactory& workloadFactory,
450  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
451  const armnn::ITensorHandleFactory& tensorHandleFactory);
452 
453 template LayerTestResult<int32_t, 3>
454 ArgMinWidthTest<armnn::DataType::Signed32>(
455  armnn::IWorkloadFactory& workloadFactory,
456  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
457  const armnn::ITensorHandleFactory& tensorHandleFactory);
458 
459 template LayerTestResult<int32_t, 3>
460 ArgMinWidthTest<armnn::DataType::QAsymmS8>(
461  armnn::IWorkloadFactory& workloadFactory,
462  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
463  const armnn::ITensorHandleFactory& tensorHandleFactory);
464 
465 template LayerTestResult<int32_t, 3>
466 ArgMinWidthTest<armnn::DataType::QAsymmU8>(
467  armnn::IWorkloadFactory& workloadFactory,
468  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
469  const armnn::ITensorHandleFactory& tensorHandleFactory);
TensorCopyUtils.hpp
armnn::TensorInfo::GetShape
const TensorShape & GetShape() const
Definition: Tensor.hpp:191
ArgMaxSimpleTest
LayerTestResult< int32_t, 3 > ArgMaxSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:65
armnn::IWorkloadFactory::CreateArgMinMax
virtual std::unique_ptr< IWorkload > CreateArgMinMax(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info) const
Definition: WorkloadFactory.cpp:1519
WorkloadTestUtils.hpp
armnn::ArgMinMaxDescriptor::m_Function
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Definition: Descriptors.hpp:70
ArgMinMaxTestImpl.hpp
ArgMaxHeightTest
LayerTestResult< int32_t, 3 > ArgMaxHeightTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:193
TensorHelpers.hpp
ArgMinChannelTest
LayerTestResult< int32_t, 3 > ArgMinChannelTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:121
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition: IBackendInternal.hpp:99
armnn::TensorInfo::SetQuantizationScale
void SetQuantizationScale(float scale)
Definition: Tensor.cpp:475
CopyDataFromITensorHandle
void CopyDataFromITensorHandle(void *memory, const armnn::ITensorHandle *tensorHandle)
Definition: TensorCopyUtils.cpp:14
armnn::ArgMinMaxFunction
ArgMinMaxFunction
Definition: Types.hpp:89
DataTypeUtils.hpp
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
armnn::ArgMinMaxDescriptor::m_Axis
int m_Axis
Axis to reduce across the input tensor.
Definition: Descriptors.hpp:72
armnn::TensorInfo::SetQuantizationOffset
void SetQuantizationOffset(int32_t offset)
Definition: Tensor.cpp:491
ArgMinSimpleTest
LayerTestResult< int32_t, 3 > ArgMinSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:93
armnn::ITensorHandleFactory::CreateTensorHandle
virtual std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo) const =0
ArgMinWidthTest
LayerTestResult< int32_t, 3 > ArgMinWidthTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:229
armnn::TensorInfo::GetNumElements
unsigned int GetNumElements() const
Definition: Tensor.hpp:196
CopyDataToITensorHandle
void CopyDataToITensorHandle(armnn::ITensorHandle *tensorHandle, const void *memory)
Definition: TensorCopyUtils.cpp:9
ArgMaxChannelTest
LayerTestResult< int32_t, 3 > ArgMaxChannelTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:157