blob: beab583cabea742639e0255df85253e73857b58b [file] [log] [blame]
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
Sadik Armagana097d2a2021-11-24 15:47:28 +00008#include <armnnTestUtils/LayerTestResult.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +01009
Narumol Prangnawarat403a1852020-03-12 14:24:13 +000010#include <BFloat16.hpp>
11
Matteo Martincighe5b8eb92019-11-28 15:45:42 +000012#include <armnn/backends/IBackendInternal.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010013#include <backendsCommon/WorkloadFactory.hpp>
14
15LayerTestResult<float, 4> Debug4dFloat32Test(
16 armnn::IWorkloadFactory& workloadFactory,
17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
18
19LayerTestResult<float, 3> Debug3dFloat32Test(
20 armnn::IWorkloadFactory& workloadFactory,
21 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
22
23LayerTestResult<float, 2> Debug2dFloat32Test(
24 armnn::IWorkloadFactory& workloadFactory,
25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
26
27LayerTestResult<float, 1> Debug1dFloat32Test(
28 armnn::IWorkloadFactory& workloadFactory,
29 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
30
Narumol Prangnawarat403a1852020-03-12 14:24:13 +000031LayerTestResult<armnn::BFloat16, 4> Debug4dBFloat16Test(
32 armnn::IWorkloadFactory& workloadFactory,
33 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
34
35LayerTestResult<armnn::BFloat16, 3> Debug3dBFloat16Test(
36 armnn::IWorkloadFactory& workloadFactory,
37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
38
39LayerTestResult<armnn::BFloat16, 2> Debug2dBFloat16Test(
40 armnn::IWorkloadFactory& workloadFactory,
41 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
42
43LayerTestResult<armnn::BFloat16, 1> Debug1dBFloat16Test(
44 armnn::IWorkloadFactory& workloadFactory,
45 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
46
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010047LayerTestResult<uint8_t, 4> Debug4dUint8Test(
48 armnn::IWorkloadFactory& workloadFactory,
49 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
50
51LayerTestResult<uint8_t, 3> Debug3dUint8Test(
52 armnn::IWorkloadFactory& workloadFactory,
53 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
54
55LayerTestResult<uint8_t, 2> Debug2dUint8Test(
56 armnn::IWorkloadFactory& workloadFactory,
57 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
58
59LayerTestResult<uint8_t, 1> Debug1dUint8Test(
60 armnn::IWorkloadFactory& workloadFactory,
61 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
62
63LayerTestResult<int16_t, 4> Debug4dInt16Test(
64 armnn::IWorkloadFactory& workloadFactory,
65 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
66
67LayerTestResult<int16_t, 3> Debug3dInt16Test(
68 armnn::IWorkloadFactory& workloadFactory,
69 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
70
71LayerTestResult<int16_t, 2> Debug2dInt16Test(
72 armnn::IWorkloadFactory& workloadFactory,
73 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
74
75LayerTestResult<int16_t, 1> Debug1dInt16Test(
76 armnn::IWorkloadFactory& workloadFactory,
77 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);