1 // 2 // Copyright © 2019 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "LayerTestResult.hpp" 9 10 #include <Half.hpp> 11 12 #include <armnn/backends/IBackendInternal.hpp> 13 #include <backendsCommon/WorkloadFactory.hpp> 14 15 LayerTestResult<float, 4> StackAxis0Float32Test( 16 armnn::IWorkloadFactory& workloadFactory, 17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 18 const armnn::ITensorHandleFactory& tensorHandleFactory); 19 20 LayerTestResult<float, 4> StackOutput4DAxis1Float32Test( 21 armnn::IWorkloadFactory& workloadFactory, 22 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 23 const armnn::ITensorHandleFactory& tensorHandleFactory); 24 25 LayerTestResult<float, 4> StackOutput4DAxis2Float32Test( 26 armnn::IWorkloadFactory& workloadFactory, 27 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 28 const armnn::ITensorHandleFactory& tensorHandleFactory); 29 30 LayerTestResult<float, 4> StackOutput4DAxis3Float32Test( 31 armnn::IWorkloadFactory& workloadFactory, 32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 33 const armnn::ITensorHandleFactory& tensorHandleFactory); 34 35 LayerTestResult<float, 3> StackOutput3DInputs3Float32Test( 36 armnn::IWorkloadFactory& workloadFactory, 37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 38 const armnn::ITensorHandleFactory& tensorHandleFactory); 39 40 LayerTestResult<float, 5> StackOutput5DFloat32Test( 41 armnn::IWorkloadFactory& workloadFactory, 42 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 43 const armnn::ITensorHandleFactory& tensorHandleFactory); 44 45 LayerTestResult<armnn::Half, 4> StackFloat16Test( 46 armnn::IWorkloadFactory& workloadFactory, 47 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 48 const armnn::ITensorHandleFactory& tensorHandleFactory); 49