Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 6a1d506cb0619c6bcf8685ada56ddf4507c2f2d4 / . / src / backends / backendsCommon / test / layerTests / DebugTestImpl.hpp
blob: 250c658cbf39cf4060613301fb5c11ab3f5e40c3 [file] [log] [blame]
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]1//
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 +0000[diff] [blame]8#include <armnnTestUtils/LayerTestResult.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]9
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]10#include <BFloat16.hpp>
11
Matteo Martincighe5b8eb92019-11-28 15:45:42 +0000[diff] [blame]12#include <armnn/backends/IBackendInternal.hpp>
Colm Donelan0c479742021-12-10 12:43:54 +0000[diff] [blame]13#include <armnn/backends/WorkloadFactory.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]14
15LayerTestResult<float, 4> Debug4dFloat32Test(
16 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
18 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]19
20LayerTestResult<float, 3> Debug3dFloat32Test(
21 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]22 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
23 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]24
25LayerTestResult<float, 2> Debug2dFloat32Test(
26 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]27 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
28 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]29
30LayerTestResult<float, 1> Debug1dFloat32Test(
31 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
33 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]34
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]35LayerTestResult<armnn::BFloat16, 4> Debug4dBFloat16Test(
36 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
38 bool toFile);
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]39
40LayerTestResult<armnn::BFloat16, 3> Debug3dBFloat16Test(
41 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]42 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
43 bool toFile);
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]44
45LayerTestResult<armnn::BFloat16, 2> Debug2dBFloat16Test(
46 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]47 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
48 bool toFile);
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]49
50LayerTestResult<armnn::BFloat16, 1> Debug1dBFloat16Test(
51 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]52 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
53 bool toFile);
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]54
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]55LayerTestResult<uint8_t, 4> Debug4dUint8Test(
56 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]57 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
58 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]59
60LayerTestResult<uint8_t, 3> Debug3dUint8Test(
61 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]62 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
63 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]64
65LayerTestResult<uint8_t, 2> Debug2dUint8Test(
66 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]67 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
68 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]69
70LayerTestResult<uint8_t, 1> Debug1dUint8Test(
71 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]72 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
73 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]74
75LayerTestResult<int16_t, 4> Debug4dInt16Test(
76 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]77 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
78 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]79
80LayerTestResult<int16_t, 3> Debug3dInt16Test(
81 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]82 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
83 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]84
85LayerTestResult<int16_t, 2> Debug2dInt16Test(
86 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]87 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
88 bool toFile);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]89
90LayerTestResult<int16_t, 1> Debug1dInt16Test(
91 armnn::IWorkloadFactory& workloadFactory,
Keith Davis15f9c682022-10-14 15:50:33 +0100[diff] [blame]92 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
93 bool toFile);