1 // 2 // Copyright © 2019 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnnTestUtils/LayerTestResult.hpp> 9 10 #include <ResolveType.hpp> 11 12 #include <armnn/backends/IBackendInternal.hpp> 13 #include <armnn/backends/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);