1 // 2 // Copyright © 2017 Arm Ltd. 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, armnn::DataType ArmnnBType, typename T = armnn::ResolveType<ArmnnType>> 16 LayerTestResult<T, 4> SimpleTransposeConvolution2dTest( 17 armnn::IWorkloadFactory& workloadFactory, 18 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 19 const armnn::ITensorHandleFactory& tensorHandleFactory, 20 bool biasEnabled, 21 const armnn::DataLayout layout); 22 23 template<armnn::DataType ArmnnType, armnn::DataType ArmnnBType, typename T = armnn::ResolveType<ArmnnType>> 24 LayerTestResult<T, 4> PaddedTransposeConvolution2dTest( 25 armnn::IWorkloadFactory& workloadFactory, 26 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 27 const armnn::ITensorHandleFactory& tensorHandleFactory, 28 bool biasEnabled, 29 const armnn::DataLayout layout); 30 31 template<armnn::DataType ArmnnType, armnn::DataType ArmnnBType, typename T = armnn::ResolveType<ArmnnType>> 32 LayerTestResult<T, 4> StridedTransposeConvolution2dTest( 33 armnn::IWorkloadFactory& workloadFactory, 34 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 35 const armnn::ITensorHandleFactory& tensorHandleFactory, 36 bool biasEnabled, 37 const armnn::DataLayout layout); 38 39 template<armnn::DataType ArmnnType, armnn::DataType ArmnnBType, typename T = armnn::ResolveType<ArmnnType>> 40 LayerTestResult<T, 4> MultiChannelTransposeConvolution2dTest( 41 armnn::IWorkloadFactory& workloadFactory, 42 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 43 const armnn::ITensorHandleFactory& tensorHandleFactory, 44 const armnn::DataLayout layout); 45 46 LayerTestResult<uint8_t, 4> TransposeConvolution2dPerAxisQuantTest( 47 armnn::IWorkloadFactory& workloadFactory, 48 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 49 const armnn::ITensorHandleFactory& tensorHandleFactory, 50 const armnn::DataLayout layout); 51