1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include <armnnTestUtils/LayerTestResult.hpp> 8 9 #include <Half.hpp> 10 11 #include <armnn/backends/IBackendInternal.hpp> 12 #include <armnn/backends/WorkloadFactory.hpp> 13 14 LayerTestResult<uint8_t, 4> SpaceToDepthNchwAsymmQ8Test( 15 armnn::IWorkloadFactory& workloadFactory, 16 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 17 const armnn::ITensorHandleFactory& tensorHandleFactory); 18 19 LayerTestResult<uint8_t, 4> SpaceToDepthNhwcAsymmQ8Test( 20 armnn::IWorkloadFactory& workloadFactory, 21 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 22 const armnn::ITensorHandleFactory& tensorHandleFactory); 23 24 LayerTestResult<armnn::Half, 4> SpaceToDepthNchwFloat16Test( 25 armnn::IWorkloadFactory& workloadFactory, 26 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 27 const armnn::ITensorHandleFactory& tensorHandleFactory); 28 29 LayerTestResult<armnn::Half, 4> SpaceToDepthNhwcFloat16Test( 30 armnn::IWorkloadFactory& workloadFactory, 31 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 32 const armnn::ITensorHandleFactory& tensorHandleFactory); 33 34 LayerTestResult<float, 4> SpaceToDepthNhwcFloat32Test1( 35 armnn::IWorkloadFactory& workloadFactory, 36 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 37 const armnn::ITensorHandleFactory& tensorHandleFactory); 38 39 LayerTestResult<float, 4> SpaceToDepthNchwFloat32Test1( 40 armnn::IWorkloadFactory& workloadFactory, 41 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 42 const armnn::ITensorHandleFactory& tensorHandleFactory); 43 44 LayerTestResult<float, 4> SpaceToDepthNhwcFloat32Test2( 45 armnn::IWorkloadFactory& workloadFactory, 46 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 47 const armnn::ITensorHandleFactory& tensorHandleFactory); 48 49 LayerTestResult<float, 4> SpaceToDepthNchwFloat32Test2( 50 armnn::IWorkloadFactory& workloadFactory, 51 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 52 const armnn::ITensorHandleFactory& tensorHandleFactory); 53 54 LayerTestResult<int16_t, 4> SpaceToDepthNhwcQSymm16Test( 55 armnn::IWorkloadFactory& workloadFactory, 56 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 57 const armnn::ITensorHandleFactory& tensorHandleFactory); 58 59 LayerTestResult<int16_t, 4> SpaceToDepthNchwQSymm16Test( 60 armnn::IWorkloadFactory& workloadFactory, 61 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 62 const armnn::ITensorHandleFactory& tensorHandleFactory); 63