1 // 2 // Copyright © 2020 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<typename T , std::size_t n> 16 LayerTestResult<int32_t, 1> RankTest( 17 armnn::TensorInfo inputTensorInfo, 18 std::vector<T> input, 19 armnn::IWorkloadFactory& workloadFactory, 20 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 21 const armnn::ITensorHandleFactory& tensorHandleFactory); 22 23 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 24 LayerTestResult<int32_t, 1> RankDimSize1Test( 25 armnn::IWorkloadFactory& workloadFactory, 26 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 27 const armnn::ITensorHandleFactory& tensorHandleFactory); 28 29 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 30 LayerTestResult<int32_t, 1> RankDimSize2Test( 31 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, 1> RankDimSize3Test( 37 armnn::IWorkloadFactory& workloadFactory, 38 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 39 const armnn::ITensorHandleFactory& tensorHandleFactory); 40 41 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 42 LayerTestResult<int32_t, 1> RankDimSize4Test( 43 armnn::IWorkloadFactory& workloadFactory, 44 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 45 const armnn::ITensorHandleFactory& tensorHandleFactory); 46