Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 826a543547f325bfac7da507d19a55af9c138a50 / . / src / backends / backendsCommon / test / layerTests / PadTestImpl.cpp
blob: 5511f86e17320165c9ff47f814a97118c7e41bec [file] [log] [blame]
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]1//
Teresa Charlinfbf0e5b2020-08-17 01:01:06 +0100[diff] [blame]2// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
4//
5
6#include "PadTestImpl.hpp"
7
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]8#include <QuantizeHelper.hpp>
9
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]10#include <backendsCommon/test/TensorCopyUtils.hpp>
11#include <backendsCommon/test/WorkloadTestUtils.hpp>
12
13#include <test/TensorHelpers.hpp>
14
15//
16// Implementation templates
17//
18
19template<armnn::DataType ArmnnType, typename T>
20LayerTestResult<T, 2> Pad2dTestCommon(
21 armnn::IWorkloadFactory& workloadFactory,
22 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]23 const armnn::ITensorHandleFactory& tensorHandleFactory,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]24 float qScale,
25 int32_t qOffset,
26 const float customPaddingValue)
27{
Jan Eilers8eb25602020-03-09 12:13:48 +0000[diff] [blame]28 IgnoreUnused(memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]29 const armnn::TensorShape inputShape{ 3, 3 };
30 const armnn::TensorShape outputShape{ 7, 7 };
31
32 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset);
33 const armnn::TensorInfo outputTensorInfo(outputShape, ArmnnType, qScale, qOffset);
34
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]35 std::vector<T> inputValues = armnnUtils::QuantizedVector<T>(
36 {
37 // Height (3) x Width (3)
38 4, 8, 6,
39 7, 4, 4,
40 3, 2, 4
41 },
42 qScale, qOffset);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]43
44 auto p = customPaddingValue;
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]45 std::vector<T> expectedOutputValues = armnnUtils::QuantizedVector<T>(
46 {
47 p, p, p, p, p, p, p,
48 p, p, p, p, p, p, p,
49 p, p, 4, 8, 6, p, p,
50 p, p, 7, 4, 4, p, p,
51 p, p, 3, 2, 4, p, p,
52 p, p, p, p, p, p, p,
53 p, p, p, p, p, p, p
54 },
55 qScale, qOffset);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]56
57 auto inputTensor = MakeTensor<T, 2>(inputTensorInfo, std::vector<T>(inputValues));
58
59 LayerTestResult<T, 2> result(outputTensorInfo);
60 result.outputExpected = MakeTensor<T, 2>(outputTensorInfo, std::vector<T>(expectedOutputValues));
61
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]62 std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
63 std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
64
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]65
66 armnn::PadQueueDescriptor descriptor;
67
68 std::vector<std::pair<unsigned int, unsigned int>> padList;
69 padList.push_back(std::pair<unsigned int, unsigned int>(2,2));
70 padList.push_back(std::pair<unsigned int, unsigned int>(2,2));
71
72 descriptor.m_Parameters.m_PadList = padList;
73 descriptor.m_Parameters.m_PadValue = customPaddingValue;
74 armnn::WorkloadInfo info;
75
76 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get());
77 AddOutputToWorkload(descriptor, info, outputTensorInfo, outputHandle.get());
78
79 std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreatePad(descriptor, info);
80
81 inputHandle->Allocate();
82 outputHandle->Allocate();
83
84 CopyDataToITensorHandle(inputHandle.get(), &inputTensor[0][0]);
85
86 workload->PostAllocationConfigure();
87 workload->Execute();
88
89 CopyDataFromITensorHandle(&result.output[0][0], outputHandle.get());
90
91 return result;
92}
93
94template<armnn::DataType ArmnnType, typename T>
95LayerTestResult<T, 3> Pad3dTestCommon(
96 armnn::IWorkloadFactory& workloadFactory,
97 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]98 const armnn::ITensorHandleFactory& tensorHandleFactory,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]99 float qScale,
100 int32_t qOffset)
101{
Jan Eilers8eb25602020-03-09 12:13:48 +0000[diff] [blame]102 IgnoreUnused(memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]103 const armnn::TensorShape inputShape{ 2, 2, 2 };
104 const armnn::TensorShape outputShape{ 3, 5, 6 };
105
106 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset);
107 const armnn::TensorInfo outputTensorInfo(outputShape, ArmnnType, qScale, qOffset);
108
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]109 std::vector<T> inputValues = armnnUtils::QuantizedVector<T>(
110 {
111 // Channel 0, Height (2) x Width (2)
112 0, 4,
113 2, 5,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]114
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]115 // Channel 1, Height (2) x Width (2)
116 6, 1,
117 5, 2
118 },
119 qScale, qOffset);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]120
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]121 std::vector<T> expectedOutputValues = armnnUtils::QuantizedVector<T>(
122 {
123 0, 0, 0, 0, 0, 0,
124 0, 0, 0, 0, 0, 0,
125 0, 0, 0, 4, 0, 0,
126 0, 0, 2, 5, 0, 0,
127 0, 0, 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]128
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]129 0, 0, 0, 0, 0, 0,
130 0, 0, 0, 0, 0, 0,
131 0, 0, 6, 1, 0, 0,
132 0, 0, 5, 2, 0, 0,
133 0, 0, 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]134
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]135 0, 0, 0, 0, 0, 0,
136 0, 0, 0, 0, 0, 0,
137 0, 0, 0, 0, 0, 0,
138 0, 0, 0, 0, 0, 0,
139 0, 0, 0, 0, 0, 0
140 },
141 qScale, qOffset);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]142
143 auto inputTensor = MakeTensor<T, 3>(inputTensorInfo, std::vector<T>(inputValues));
144
145 LayerTestResult<T, 3> result(outputTensorInfo);
146 result.outputExpected = MakeTensor<T, 3>(outputTensorInfo, std::vector<T>(expectedOutputValues));
147
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]148 std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
149 std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
150
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]151
152 armnn::PadQueueDescriptor descriptor;
153
154 std::vector<std::pair<unsigned int, unsigned int>> PadList;
155 PadList.push_back(std::pair<unsigned int, unsigned int>(0,1));
156 PadList.push_back(std::pair<unsigned int, unsigned int>(2,1));
157 PadList.push_back(std::pair<unsigned int, unsigned int>(2,2));
158
159 descriptor.m_Parameters.m_PadList = PadList;
160 armnn::WorkloadInfo info;
161
162 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get());
163 AddOutputToWorkload(descriptor, info, outputTensorInfo, outputHandle.get());
164
165 std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreatePad(descriptor, info);
166
167 inputHandle->Allocate();
168 outputHandle->Allocate();
169
170 CopyDataToITensorHandle(inputHandle.get(), &inputTensor[0][0][0]);
171
172 workload->PostAllocationConfigure();
173 workload->Execute();
174
175 CopyDataFromITensorHandle(&result.output[0][0][0], outputHandle.get());
176
177 return result;
178}
179
180template<armnn::DataType ArmnnType, typename T>
181LayerTestResult<T, 4> Pad4dTestCommon(
182 armnn::IWorkloadFactory& workloadFactory,
183 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]184 const armnn::ITensorHandleFactory& tensorHandleFactory,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]185 float qScale,
186 int32_t qOffset)
187{
Jan Eilers8eb25602020-03-09 12:13:48 +0000[diff] [blame]188 IgnoreUnused(memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]189 const armnn::TensorShape inputShape{ 2, 2, 3, 2 };
190 const armnn::TensorShape outputShape{ 4, 5, 7, 4 };
191
192 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset);
193 const armnn::TensorInfo outputTensorInfo(outputShape, ArmnnType, qScale, qOffset);
194
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]195 std::vector<T> inputValues = armnnUtils::QuantizedVector<T>(
196 {
197 // Batch 0, Channel 0, Height (3) x Width (2)
198 0, 1,
199 2, 3,
200 4, 5,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]201
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]202 // Batch 0, Channel 1, Height (3) x Width (2)
203 6, 7,
204 8, 9,
205 10, 11,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]206
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]207 // Batch 1, Channel 0, Height (3) x Width (2)
208 12, 13,
209 14, 15,
210 16, 17,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]211
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]212 // Batch 1, Channel 1, Height (3) x Width (2)
213 18, 19,
214 20, 21,
215 22, 23
216 },
217 qScale, qOffset);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]218
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]219 std::vector<T> expectedOutputValues = armnnUtils::QuantizedVector<T>(
220 {
221 0, 0, 0, 0,
222 0, 0, 0, 0,
223 0, 0, 0, 0,
224 0, 0, 0, 0,
225 0, 0, 0, 0,
226 0, 0, 0, 0,
227 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]228
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]229 0, 0, 0, 0,
230 0, 0, 0, 0,
231 0, 0, 0, 0,
232 0, 0, 0, 0,
233 0, 0, 0, 0,
234 0, 0, 0, 0,
235 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]236
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]237 0, 0, 0, 0,
238 0, 0, 0, 0,
239 0, 0, 0, 0,
240 0, 0, 0, 0,
241 0, 0, 0, 0,
242 0, 0, 0, 0,
243 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]244
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]245 0, 0, 0, 0,
246 0, 0, 0, 0,
247 0, 0, 0, 0,
248 0, 0, 0, 0,
249 0, 0, 0, 0,
250 0, 0, 0, 0,
251 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]252
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]253 0, 0, 0, 0,
254 0, 0, 0, 0,
255 0, 0, 0, 0,
256 0, 0, 0, 0,
257 0, 0, 0, 0,
258 0, 0, 0, 0,
259 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]260
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]261 0, 0, 0, 0,
262 0, 0, 0, 0,
263 0, 0, 0, 0,
264 0, 0, 0, 0,
265 0, 0, 0, 0,
266 0, 0, 0, 0,
267 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]268
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]269 0, 0, 0, 0,
270 0, 0, 0, 0,
271 0, 0, 0, 0,
272 0, 0, 0, 0,
273 0, 0, 0, 0,
274 0, 0, 0, 0,
275 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]276
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]277 0, 0, 0, 0,
278 0, 0, 0, 0,
279 0, 0, 0, 0,
280 0, 0, 1, 0,
281 0, 2, 3, 0,
282 0, 4, 5, 0,
283 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]284
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]285 0, 0, 0, 0,
286 0, 0, 0, 0,
287 0, 0, 0, 0,
288 0, 6, 7, 0,
289 0, 8, 9, 0,
290 0, 10, 11, 0,
291 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]292
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]293 0, 0, 0, 0,
294 0, 0, 0, 0,
295 0, 0, 0, 0,
296 0, 0, 0, 0,
297 0, 0, 0, 0,
298 0, 0, 0, 0,
299 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]300
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]301 0, 0, 0, 0,
302 0, 0, 0, 0,
303 0, 0, 0, 0,
304 0, 0, 0, 0,
305 0, 0, 0, 0,
306 0, 0, 0, 0,
307 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]308
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]309 0, 0, 0, 0,
310 0, 0, 0, 0,
311 0, 0, 0, 0,
312 0, 0, 0, 0,
313 0, 0, 0, 0,
314 0, 0, 0, 0,
315 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]316
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]317 0, 0, 0, 0,
318 0, 0, 0, 0,
319 0, 0, 0, 0,
320 0, 12, 13, 0,
321 0, 14, 15, 0,
322 0, 16, 17, 0,
323 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]324
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]325 0, 0, 0, 0,
326 0, 0, 0, 0,
327 0, 0, 0, 0,
328 0, 18, 19, 0,
329 0, 20, 21, 0,
330 0, 22, 23, 0,
331 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]332
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]333 0, 0, 0, 0,
334 0, 0, 0, 0,
335 0, 0, 0, 0,
336 0, 0, 0, 0,
337 0, 0, 0, 0,
338 0, 0, 0, 0,
339 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]340
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]341 0, 0, 0, 0,
342 0, 0, 0, 0,
343 0, 0, 0, 0,
344 0, 0, 0, 0,
345 0, 0, 0, 0,
346 0, 0, 0, 0,
347 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]348
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]349 0, 0, 0, 0,
350 0, 0, 0, 0,
351 0, 0, 0, 0,
352 0, 0, 0, 0,
353 0, 0, 0, 0,
354 0, 0, 0, 0,
355 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]356
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]357 0, 0, 0, 0,
358 0, 0, 0, 0,
359 0, 0, 0, 0,
360 0, 0, 0, 0,
361 0, 0, 0, 0,
362 0, 0, 0, 0,
363 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]364
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]365 0, 0, 0, 0,
366 0, 0, 0, 0,
367 0, 0, 0, 0,
368 0, 0, 0, 0,
369 0, 0, 0, 0,
370 0, 0, 0, 0,
371 0, 0, 0, 0,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]372
Aron Virginas-Tar48623a02019-10-22 10:00:28 +0100[diff] [blame]373 0, 0, 0, 0,
374 0, 0, 0, 0,
375 0, 0, 0, 0,
376 0, 0, 0, 0,
377 0, 0, 0, 0,
378 0, 0, 0, 0,
379 0, 0, 0, 0
380 },
381 qScale, qOffset);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]382
383 auto inputTensor = MakeTensor<T, 4>(inputTensorInfo, std::vector<T>(inputValues));
384
385 LayerTestResult<T, 4> result(outputTensorInfo);
386 result.outputExpected = MakeTensor<T, 4>(outputTensorInfo, std::vector<T>(expectedOutputValues));
387
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]388 std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
389 std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]390
391 armnn::PadQueueDescriptor descriptor;
392
393 std::vector<std::pair<unsigned int, unsigned int>> PadList;
394 PadList.push_back(std::pair<unsigned int, unsigned int>(1,1));
395 PadList.push_back(std::pair<unsigned int, unsigned int>(2,1));
396 PadList.push_back(std::pair<unsigned int, unsigned int>(3,1));
397 PadList.push_back(std::pair<unsigned int, unsigned int>(1,1));
398
399 descriptor.m_Parameters.m_PadList = PadList;
400 armnn::WorkloadInfo info;
401
402 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get());
403 AddOutputToWorkload(descriptor, info, outputTensorInfo, outputHandle.get());
404
405 std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreatePad(descriptor, info);
406
407 inputHandle->Allocate();
408 outputHandle->Allocate();
409
410 CopyDataToITensorHandle(inputHandle.get(), &inputTensor[0][0][0][0]);
411
412 workload->PostAllocationConfigure();
413 workload->Execute();
414
415 CopyDataFromITensorHandle(&result.output[0][0][0][0], outputHandle.get());
416
417 return result;
418}
419
420//
421// Explicit template specializations
422//
423
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]424template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 2>
425Pad2dTestCommon<armnn::DataType::QSymmS16>(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]426 armnn::IWorkloadFactory& workloadFactory,
427 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]428 const armnn::ITensorHandleFactory& tensorHandleFactory,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]429 float qScale,
430 int32_t qOffset,
431 const float customPaddingValue);
432
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]433template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 3>
434Pad3dTestCommon<armnn::DataType::QSymmS16>(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]435 armnn::IWorkloadFactory& workloadFactory,
436 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]437 const armnn::ITensorHandleFactory& tensorHandleFactory,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]438 float qScale,
439 int32_t qOffset);
440
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]441template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 4>
442Pad4dTestCommon<armnn::DataType::QSymmS16>(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]443 armnn::IWorkloadFactory& workloadFactory,
444 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]445 const armnn::ITensorHandleFactory& tensorHandleFactory,
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]446 float qScale,
447 int32_t qOffset);
448
449//
450// Implementation functions
451//
452
453LayerTestResult<uint8_t, 2> PadUint82dTest(
454 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]455 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
456 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]457{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]458 return Pad2dTestCommon<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]459}
460
461LayerTestResult<uint8_t, 2> PadUint82dCustomPaddingTest(
462 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]463 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
464 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]465{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]466 return Pad2dTestCommon<armnn::DataType::QAsymmU8>(
467 workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0, 1.0f);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]468}
469
470LayerTestResult<uint8_t, 3> PadUint83dTest(
471 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]472 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
473 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]474{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]475 return Pad3dTestCommon<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]476}
477
478LayerTestResult<uint8_t, 4> PadUint84dTest(
479 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]480 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
481 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]482{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]483 return Pad4dTestCommon<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]484}
485
486LayerTestResult<float, 2> PadFloat322dTest(
487 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]488 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
489 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]490{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]491 return Pad2dTestCommon<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]492}
493
494LayerTestResult<float, 2> PadFloat322dCustomPaddingTest(
495 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]496 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
497 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]498{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]499 return Pad2dTestCommon<armnn::DataType::Float32>(
500 workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0, 1.0f);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]501}
502
503LayerTestResult<float, 3> PadFloat323dTest(
504 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]505 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
506 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]507{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]508 return Pad3dTestCommon<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]509}
510
511LayerTestResult<float, 4> PadFloat324dTest(
512 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]513 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
514 const armnn::ITensorHandleFactory& tensorHandleFactory)
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]515{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]516 return Pad4dTestCommon<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]517}
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]518
519LayerTestResult<armnn::BFloat16, 2> PadBFloat162dTest(
520 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]521 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
522 const armnn::ITensorHandleFactory& tensorHandleFactory)
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]523{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]524 return Pad2dTestCommon<armnn::DataType::BFloat16>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0);
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]525}
526
527LayerTestResult<armnn::BFloat16, 2> PadBFloat162dCustomPaddingTest(
528 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]529 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
530 const armnn::ITensorHandleFactory& tensorHandleFactory)
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]531{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]532 return Pad2dTestCommon<armnn::DataType::BFloat16>(
533 workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0, 1.0f);
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]534}
535
536LayerTestResult<armnn::BFloat16, 3> PadBFloat163dTest(
537 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]538 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
539 const armnn::ITensorHandleFactory& tensorHandleFactory)
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]540{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]541 return Pad3dTestCommon<armnn::DataType::BFloat16>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0);
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]542}
543
544LayerTestResult<armnn::BFloat16, 4> PadBFloat164dTest(
545 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]546 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
547 const armnn::ITensorHandleFactory& tensorHandleFactory)
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]548{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]549 return Pad4dTestCommon<armnn::DataType::BFloat16>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0);
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]550}
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]551
552LayerTestResult<int8_t, 2> PadInt82dTest(
553 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]554 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
555 const armnn::ITensorHandleFactory& tensorHandleFactory)
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]556{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]557 return Pad2dTestCommon<armnn::DataType::QSymmS8>(
558 workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0);
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]559}
560
561LayerTestResult<int8_t, 2> PadInt82dCustomPaddingTest(
562 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]563 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
564 const armnn::ITensorHandleFactory& tensorHandleFactory)
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]565{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]566 return Pad2dTestCommon<armnn::DataType::QSymmS8>(
567 workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0, 1.0f);
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]568}
569
570LayerTestResult<int8_t, 3> PadInt83dTest(
571 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]572 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
573 const armnn::ITensorHandleFactory& tensorHandleFactory)
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]574{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]575 return Pad3dTestCommon<armnn::DataType::QSymmS8>(workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0);
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]576}
577
578LayerTestResult<int8_t, 4> PadInt84dTest(
579 armnn::IWorkloadFactory& workloadFactory,
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]580 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
581 const armnn::ITensorHandleFactory& tensorHandleFactory)
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]582{
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame^]583 return Pad4dTestCommon<armnn::DataType::QSymmS8>(workloadFactory, memoryManager, tensorHandleFactory, 1.0f, 0);
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]584}