1 // 2 // Copyright © 2019 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include <armnnTestUtils/LayerTestResult.hpp> 8 9 #include <ResolveType.hpp> 10 11 #include <armnn/backends/IBackendInternal.hpp> 12 #include <armnn/backends/WorkloadFactory.hpp> 13 14 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 15 LayerTestResult<T, 4> DepthToSpaceTest1( 16 armnn::IWorkloadFactory& workloadFactory, 17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 18 armnn::DataLayout dataLayout); 19 20 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 21 LayerTestResult<T, 4> DepthToSpaceTest2( 22 armnn::IWorkloadFactory& workloadFactory, 23 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 24 armnn::DataLayout dataLayout); 25 26 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 27 LayerTestResult<T, 4> DepthToSpaceTest3( 28 armnn::IWorkloadFactory& workloadFactory, 29 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 30 armnn::DataLayout dataLayout); 31 32 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 33 LayerTestResult<T, 4> DepthToSpaceTest4( 34 armnn::IWorkloadFactory& workloadFactory, 35 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 36 armnn::DataLayout dataLayout); 37