ArmNN
 20.05
ArgMinMaxTestImpl.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2019 Arm Ltd. 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 
19 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
20 LayerTestResult<int32_t, 3> ArgMinSimpleTest(armnn::IWorkloadFactory& workloadFactory,
21  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
22 
23 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
24 LayerTestResult<int32_t, 3> ArgMinChannelTest(armnn::IWorkloadFactory& workloadFactory,
25  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
26 
27 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
28 LayerTestResult<int32_t, 3> ArgMaxChannelTest(armnn::IWorkloadFactory& workloadFactory,
29  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
30 
31 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
32 LayerTestResult<int32_t, 3> ArgMaxHeightTest(armnn::IWorkloadFactory& workloadFactory,
33  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
34 
35 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
36 LayerTestResult<int32_t, 3> ArgMinWidthTest(armnn::IWorkloadFactory& workloadFactory,
37  const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
WorkloadFactory.hpp
ResolveType.hpp
ArgMinWidthTest
LayerTestResult< int32_t, 3 > ArgMinWidthTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition: ArgMinMaxTestImpl.cpp:223
ArgMaxSimpleTest
LayerTestResult< int32_t, 3 > ArgMaxSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition: ArgMinMaxTestImpl.cpp:64
LayerTestResult.hpp
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition: IBackendInternal.hpp:90
ArgMinSimpleTest
LayerTestResult< int32_t, 3 > ArgMinSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition: ArgMinMaxTestImpl.cpp:91
ArgMaxHeightTest
LayerTestResult< int32_t, 3 > ArgMaxHeightTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition: ArgMinMaxTestImpl.cpp:188
ArgMaxChannelTest
LayerTestResult< int32_t, 3 > ArgMaxChannelTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition: ArgMinMaxTestImpl.cpp:153
ArgMinChannelTest
LayerTestResult< int32_t, 3 > ArgMinChannelTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition: ArgMinMaxTestImpl.cpp:118