xref: /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/DebugTestImpl.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
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 <BFloat16.hpp>
11 
12 #include <armnn/backends/IBackendInternal.hpp>
13 #include <armnn/backends/WorkloadFactory.hpp>
14 
15 LayerTestResult<float, 4> Debug4dFloat32Test(
16     armnn::IWorkloadFactory& workloadFactory,
17     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
18     bool toFile);
19 
20 LayerTestResult<float, 3> Debug3dFloat32Test(
21     armnn::IWorkloadFactory& workloadFactory,
22     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
23     bool toFile);
24 
25 LayerTestResult<float, 2> Debug2dFloat32Test(
26     armnn::IWorkloadFactory& workloadFactory,
27     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
28     bool toFile);
29 
30 LayerTestResult<float, 1> Debug1dFloat32Test(
31     armnn::IWorkloadFactory& workloadFactory,
32     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
33     bool toFile);
34 
35 LayerTestResult<armnn::BFloat16, 4> Debug4dBFloat16Test(
36     armnn::IWorkloadFactory& workloadFactory,
37     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
38     bool toFile);
39 
40 LayerTestResult<armnn::BFloat16, 3> Debug3dBFloat16Test(
41     armnn::IWorkloadFactory& workloadFactory,
42     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
43     bool toFile);
44 
45 LayerTestResult<armnn::BFloat16, 2> Debug2dBFloat16Test(
46     armnn::IWorkloadFactory& workloadFactory,
47     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
48     bool toFile);
49 
50 LayerTestResult<armnn::BFloat16, 1> Debug1dBFloat16Test(
51     armnn::IWorkloadFactory& workloadFactory,
52     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
53     bool toFile);
54 
55 LayerTestResult<uint8_t, 4> Debug4dUint8Test(
56     armnn::IWorkloadFactory& workloadFactory,
57     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
58     bool toFile);
59 
60 LayerTestResult<uint8_t, 3> Debug3dUint8Test(
61     armnn::IWorkloadFactory& workloadFactory,
62     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
63     bool toFile);
64 
65 LayerTestResult<uint8_t, 2> Debug2dUint8Test(
66     armnn::IWorkloadFactory& workloadFactory,
67     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
68     bool toFile);
69 
70 LayerTestResult<uint8_t, 1> Debug1dUint8Test(
71     armnn::IWorkloadFactory& workloadFactory,
72     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
73     bool toFile);
74 
75 LayerTestResult<int16_t, 4> Debug4dInt16Test(
76     armnn::IWorkloadFactory& workloadFactory,
77     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
78     bool toFile);
79 
80 LayerTestResult<int16_t, 3> Debug3dInt16Test(
81     armnn::IWorkloadFactory& workloadFactory,
82     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
83     bool toFile);
84 
85 LayerTestResult<int16_t, 2> Debug2dInt16Test(
86     armnn::IWorkloadFactory& workloadFactory,
87     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
88     bool toFile);
89 
90 LayerTestResult<int16_t, 1> Debug1dInt16Test(
91     armnn::IWorkloadFactory& workloadFactory,
92     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
93     bool toFile);
94