ArmNN
 21.02
ArgMinMaxTestImpl.hpp
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 #pragma once
7 
8 #include "LayerTestResult.hpp"
9 
10 #include <ResolveType.hpp>
11 
12 #include <armnn/backends/IBackendInternal.hpp>
13 #include <backendsCommon/WorkloadFactory.hpp>
14 
15 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
16 LayerTestResult<int32_t, 3> ArgMaxSimpleTest(armnn::IWorkloadFactory& workloadFactory,
17  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
18  const armnn::ITensorHandleFactory& tensorHandleFactory);
19 
20 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
21 LayerTestResult<int32_t, 3> ArgMinSimpleTest(armnn::IWorkloadFactory& workloadFactory,
22  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
23  const armnn::ITensorHandleFactory& tensorHandleFactory);
24 
25 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
26 LayerTestResult<int32_t, 3> ArgMinChannelTest(armnn::IWorkloadFactory& workloadFactory,
27  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
28  const armnn::ITensorHandleFactory& tensorHandleFactory);
29 
30 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
31 LayerTestResult<int32_t, 3> ArgMaxChannelTest(armnn::IWorkloadFactory& workloadFactory,
32  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
33  const armnn::ITensorHandleFactory& tensorHandleFactory);
34 
35 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
36 LayerTestResult<int32_t, 3> ArgMaxHeightTest(armnn::IWorkloadFactory& workloadFactory,
37  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
38  const armnn::ITensorHandleFactory& tensorHandleFactory);
39 
40 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
41 LayerTestResult<int32_t, 3> ArgMinWidthTest(armnn::IWorkloadFactory& workloadFactory,
42  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
43  const armnn::ITensorHandleFactory& tensorHandleFactory);
WorkloadFactory.hpp
ResolveType.hpp
LayerTestResult.hpp
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition: IBackendInternal.hpp:92
ArgMinWidthTest
LayerTestResult< int32_t, 3 > ArgMinWidthTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:228
ArgMinSimpleTest
LayerTestResult< int32_t, 3 > ArgMinSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:92
ArgMaxChannelTest
LayerTestResult< int32_t, 3 > ArgMaxChannelTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:156
ArgMaxSimpleTest
LayerTestResult< int32_t, 3 > ArgMaxSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:64
ArgMinChannelTest
LayerTestResult< int32_t, 3 > ArgMinChannelTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:120
ArgMaxHeightTest
LayerTestResult< int32_t, 3 > ArgMaxHeightTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Definition: ArgMinMaxTestImpl.cpp:192