1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "LayerTestResult.hpp" 9 10 #include <BFloat16.hpp> 11 12 #include <armnn/backends/IBackendInternal.hpp> 13 #include <backendsCommon/WorkloadFactory.hpp> 14 15 LayerTestResult<float, 4> Debug4dFloat32Test( 16 armnn::IWorkloadFactory& workloadFactory, 17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 18 19 LayerTestResult<float, 3> Debug3dFloat32Test( 20 armnn::IWorkloadFactory& workloadFactory, 21 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 22 23 LayerTestResult<float, 2> Debug2dFloat32Test( 24 armnn::IWorkloadFactory& workloadFactory, 25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 26 27 LayerTestResult<float, 1> Debug1dFloat32Test( 28 armnn::IWorkloadFactory& workloadFactory, 29 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 30 31 LayerTestResult<armnn::BFloat16, 4> Debug4dBFloat16Test( 32 armnn::IWorkloadFactory& workloadFactory, 33 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 34 35 LayerTestResult<armnn::BFloat16, 3> Debug3dBFloat16Test( 36 armnn::IWorkloadFactory& workloadFactory, 37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 38 39 LayerTestResult<armnn::BFloat16, 2> Debug2dBFloat16Test( 40 armnn::IWorkloadFactory& workloadFactory, 41 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 42 43 LayerTestResult<armnn::BFloat16, 1> Debug1dBFloat16Test( 44 armnn::IWorkloadFactory& workloadFactory, 45 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 46 47 LayerTestResult<uint8_t, 4> Debug4dUint8Test( 48 armnn::IWorkloadFactory& workloadFactory, 49 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 50 51 LayerTestResult<uint8_t, 3> Debug3dUint8Test( 52 armnn::IWorkloadFactory& workloadFactory, 53 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 54 55 LayerTestResult<uint8_t, 2> Debug2dUint8Test( 56 armnn::IWorkloadFactory& workloadFactory, 57 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 58 59 LayerTestResult<uint8_t, 1> Debug1dUint8Test( 60 armnn::IWorkloadFactory& workloadFactory, 61 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 62 63 LayerTestResult<int16_t, 4> Debug4dInt16Test( 64 armnn::IWorkloadFactory& workloadFactory, 65 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 66 67 LayerTestResult<int16_t, 3> Debug3dInt16Test( 68 armnn::IWorkloadFactory& workloadFactory, 69 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 70 71 LayerTestResult<int16_t, 2> Debug2dInt16Test( 72 armnn::IWorkloadFactory& workloadFactory, 73 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 74 75 LayerTestResult<int16_t, 1> Debug1dInt16Test( 76 armnn::IWorkloadFactory& workloadFactory, 77 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); 78