Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 6a1d506cb0619c6bcf8685ada56ddf4507c2f2d4 / . / src / backends / backendsCommon / test / layerTests / Conv3dTestImpl.hpp
blob: e17a9193d4e9bf9d2a5fd88799149efbfd447b5f [file] [log] [blame]
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]1//
2// Copyright © 2021 Arm Ltd and Contributors. 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>
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]9
10#include <Half.hpp>
11
12#include <ResolveType.hpp>
13
14#include <armnn/Types.hpp>
15
16#include <armnn/backends/IBackendInternal.hpp>
Colm Donelan0c479742021-12-10 12:43:54 +0000[diff] [blame]17#include <armnn/backends/WorkloadFactory.hpp>
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]18
19//
20// Convolution3d
21//
22
23LayerTestResult<float, 5> SimpleConvolution3d3x3x3Float32Test(
24 armnn::IWorkloadFactory& workloadFactory,
25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
26 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]27 bool biasEnabled,
28 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]29
30LayerTestResult<int8_t , 5> SimpleConvolution3d3x3x3Int8Test(
31 armnn::IWorkloadFactory& workloadFactory,
32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
33 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]34 bool biasEnabled,
35 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]36
Rob Hugheseb70c912021-10-07 08:53:58 +0100[diff] [blame]37LayerTestResult<uint8_t, 5> SimpleConvolution3d3x3x3Uint8Test(
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]38 armnn::IWorkloadFactory& workloadFactory,
39 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
40 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]41 bool biasEnabled,
42 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]43
44LayerTestResult<int16_t, 5> SimpleConvolution3d3x3x3Int16Test(
45 armnn::IWorkloadFactory& workloadFactory,
46 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
47 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]48 bool biasEnabled,
49 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]50
51LayerTestResult<float, 5> Convolution3d2x2x2Strides3x5x5Float32Test(
52 armnn::IWorkloadFactory& workloadFactory,
53 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
54 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]55 bool biasEnabled,
56 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]57
58LayerTestResult<int8_t , 5> Convolution3d2x2x2Strides3x5x5Int8Test(
59 armnn::IWorkloadFactory& workloadFactory,
60 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
61 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]62 bool biasEnabled,
63 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]64
Rob Hugheseb70c912021-10-07 08:53:58 +0100[diff] [blame]65LayerTestResult<uint8_t, 5> Convolution3d2x2x2Strides3x5x5Uint8Test(
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]66 armnn::IWorkloadFactory& workloadFactory,
67 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
68 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]69 bool biasEnabled,
70 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]71
72LayerTestResult<int16_t, 5> Convolution3d2x2x2Strides3x5x5Int16Test(
73 armnn::IWorkloadFactory& workloadFactory,
74 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
75 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]76 bool biasEnabled,
77 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]78
79LayerTestResult<float, 5> Convolution3d2x2x2Dilation2x2x2Float32Test(
80 armnn::IWorkloadFactory& workloadFactory,
81 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
82 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]83 bool biasEnabled,
84 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]85
86LayerTestResult<int8_t , 5> Convolution3d2x2x2Dilation2x2x2Int8Test(
87 armnn::IWorkloadFactory& workloadFactory,
88 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
89 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]90 bool biasEnabled,
91 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]92
Rob Hugheseb70c912021-10-07 08:53:58 +0100[diff] [blame]93LayerTestResult<uint8_t, 5> Convolution3d2x2x2Dilation2x2x2Uint8Test(
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]94 armnn::IWorkloadFactory& workloadFactory,
95 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
96 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]97 bool biasEnabled,
98 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]99
100LayerTestResult<int16_t, 5> Convolution3d2x2x2Dilation2x2x2Int16Test(
101 armnn::IWorkloadFactory& workloadFactory,
102 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
103 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]104 bool biasEnabled,
105 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]106
107LayerTestResult<float, 5> Convolution3dPaddingSame3x3x3Float32Test(
108 armnn::IWorkloadFactory& workloadFactory,
109 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
110 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]111 bool biasEnabled,
112 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]113
114LayerTestResult<int8_t , 5> Convolution3dPaddingSame3x3x3Int8Test(
115 armnn::IWorkloadFactory& workloadFactory,
116 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
117 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]118 bool biasEnabled,
119 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]120
Rob Hugheseb70c912021-10-07 08:53:58 +0100[diff] [blame]121LayerTestResult<uint8_t, 5> Convolution3dPaddingSame3x3x3Uint8Test(
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]122 armnn::IWorkloadFactory& workloadFactory,
123 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
124 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]125 bool biasEnabled,
126 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]127
128LayerTestResult<int16_t, 5> Convolution3dPaddingSame3x3x3Int16Test(
129 armnn::IWorkloadFactory& workloadFactory,
130 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
131 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]132 bool biasEnabled,
133 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]134
135LayerTestResult<float, 5> Convolution3dStrideDilationPadding3x3x3Float32Test(
136 armnn::IWorkloadFactory& workloadFactory,
137 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
138 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]139 bool biasEnabled,
140 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]141
142LayerTestResult<float, 5> Convolution3d2x2x2Stride3x3x3SmallFloat32Test(
143 armnn::IWorkloadFactory& workloadFactory,
144 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
145 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]146 bool biasEnabled,
147 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]148
149LayerTestResult<armnn::Half, 5> Convolution3d2x3x3Float16Test(
150 armnn::IWorkloadFactory& workloadFactory,
151 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
152 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]153 bool biasEnabled,
154 armnn::DataLayout dataLayout);
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]155
156LayerTestResult<armnn::Half, 5> Convolution3d2x2x2SmallFloat16Test(
157 armnn::IWorkloadFactory& workloadFactory,
158 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
159 const armnn::ITensorHandleFactory& tensorHandleFactory,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]160 bool biasEnabled,
161 armnn::DataLayout dataLayout);