ArmNN
 22.02
Functions
BatchToSpaceNdTestImpl.hpp File Reference
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <armnn/backends/WorkloadFactory.hpp>
#include <DataTypeUtils.hpp>
#include <armnnTestUtils/LayerTestResult.hpp>
#include <armnnTestUtils/TensorCopyUtils.hpp>
#include <armnnTestUtils/TensorHelpers.hpp>
#include <armnnTestUtils/WorkloadTestUtils.hpp>

Go to the source code of this file.

Functions

template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest1 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest2 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest3 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest4 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest5 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest6 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNhwcTest7 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest1 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest2 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest3 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest4 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest5 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest6 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult< T, 4 > BatchToSpaceNdNchwTest7 (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 

Function Documentation

◆ BatchToSpaceNdNchwTest1()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest1 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 282 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

286 {
287  const unsigned int inputShape[] = {4, 3, 1, 1};
288  const unsigned int outputShape[] = {1, 3, 2, 2};
289 
290  std::vector<float> input({1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f});
291 
292  std::vector<float> expectedOutput({
293  // Batch 0, Channel 0, Height (2) x Width (2)
294  1.0f, 4.0f,
295  7.0f, 10.0f,
296 
297  // Batch 0, Channel 1, Height (2) x Width (2)
298  2.0f, 5.0f,
299  8.0f, 11.0f,
300 
301  // Batch 0, Channel 2, Height (2) x Width (2)
302  3.0f, 6.0f,
303  9.0f, 12.0f,
304  });
305 
306  std::vector<unsigned int> blockShape({2, 2});
307  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
308 
309  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
310  armnn::DataLayout::NCHW, inputShape, input, blockShape,
311  crops, outputShape, expectedOutput);
312 }

◆ BatchToSpaceNdNchwTest2()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest2 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 315 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

319 {
320  const unsigned int inputShape[] = {4, 1, 1, 1};
321  const unsigned int outputShape[] = {1, 1, 2, 2};
322 
323  std::vector<float> input({
324  // Batch 0, Height 0, Width (2) x Channel (1)
325  1.0f, 2.0f, 3.0f, 4.0f
326  });
327 
328  std::vector<float> expectedOutput({1.0f, 2.0f, 3.0f, 4.0f});
329 
330  std::vector<unsigned int> blockShape({2, 2});
331  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
332 
333  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
334  armnn::DataLayout::NCHW, inputShape, input, blockShape,
335  crops, outputShape, expectedOutput);
336 }

◆ BatchToSpaceNdNchwTest3()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest3 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 339 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

343 {
344  const unsigned int inputShape[] = {4, 3, 1, 1};
345  const unsigned int outputShape[] = {1, 3, 2, 2};
346 
347  std::vector<float> input({1.0f, 3.0f, 5.0f, 7.0f, 9.0f, 11.0f, 2.0f, 4.0f, 6.0f, 8.0f, 10.0f, 12.0f});
348 
349  std::vector<float> expectedOutput({
350  // Batch 0, Channel 0, Height (2) x Width (2)
351  1.0f, 7.0f,
352  2.0f, 8.0f,
353 
354  // Batch 0, Channel 1, Height (2) x Width (2)
355  3.0f, 9.0f,
356  4.0f, 10.0f,
357 
358  // Batch 0, Channel 2, Height (2) x Width (2)
359  5.0f, 11.0f,
360  6.0f, 12.0f,
361  });
362 
363  std::vector<unsigned int> blockShape({2, 2});
364  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
365 
366  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
367  armnn::DataLayout::NCHW, inputShape, input, blockShape,
368  crops, outputShape, expectedOutput);
369 }

◆ BatchToSpaceNdNchwTest4()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest4 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 372 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

376 {
377  const unsigned int inputShape[] = {4, 3, 1, 1};
378  const unsigned int outputShape[] = {1, 3, 2, 2};
379 
380  std::vector<float> input({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12});
381 
382  std::vector<float> expectedOutput({
383  // Batch 0, Channel 0, Height (2) x Width (2)
384  1, 4,
385  7, 10,
386 
387  // Batch 0, Channel 1, Height (2) x Width (2)
388  2, 5,
389  8, 11,
390 
391  // Batch 0, Channel 2, Height (2) x Width (2)
392  3, 6,
393  9, 12,
394  });
395 
396  std::vector<unsigned int> blockShape({2, 2});
397  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
398 
399  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
400  armnn::DataLayout::NCHW, inputShape, input, blockShape,
401  crops, outputShape, expectedOutput);
402 }

◆ BatchToSpaceNdNchwTest5()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest5 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 405 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

409 {
410  const unsigned int inputShape[] = {4, 1, 1, 1};
411  const unsigned int outputShape[] = {1, 1, 2, 2};
412 
413  std::vector<float> input({
414  // Batch 0, Height 0, Width (2) x Channel (1)
415  1, 2, 3, 4
416  });
417 
418  std::vector<float> expectedOutput({1, 2, 3, 4});
419 
420  std::vector<unsigned int> blockShape({2, 2});
421  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
422 
423  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
424  armnn::DataLayout::NCHW, inputShape, input, blockShape,
425  crops, outputShape, expectedOutput);
426 }

◆ BatchToSpaceNdNchwTest6()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest6 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 429 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

433 {
434  const unsigned int inputShape[] = {4, 3, 1, 1};
435  const unsigned int outputShape[] = {1, 3, 2, 2};
436 
437  std::vector<float> input({1, 3, 5, 7, 9, 11, 2, 4, 6, 8, 10, 12});
438 
439  std::vector<float> expectedOutput({
440  // Batch 0, Channel 0, Height (2) x Width (2)
441  1, 7,
442  2, 8,
443 
444  // Batch 0, Channel 1, Height (2) x Width (2)
445  3, 9,
446  4, 10,
447 
448  // Batch 0, Channel 2, Height (2) x Width (2)
449  5, 11,
450  6, 12,
451  });
452 
453  std::vector<unsigned int> blockShape({2, 2});
454  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
455 
456  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
457  armnn::DataLayout::NCHW, inputShape, input, blockShape,
458  crops, outputShape, expectedOutput);
459 }

◆ BatchToSpaceNdNchwTest7()

LayerTestResult<T, 4> BatchToSpaceNdNchwTest7 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 462 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NCHW.

466 {
467  const unsigned int inputShape[] = {8, 1, 1, 3};
468  const unsigned int outputShape[] = {2, 1, 2, 4};
469 
470  std::vector<float> input({
471  0, 1, 3, 0, 9, 11,
472  0, 2, 4, 0, 10, 12,
473  0, 5, 7, 0, 13, 15,
474  0, 6, 8, 0, 14, 16
475  });
476 
477  std::vector<float> expectedOutput({
478  1, 2, 3, 4,
479  5, 6, 7, 8,
480  9, 10, 11, 12,
481  13, 14, 15, 16
482  });
483 
484  std::vector<unsigned int> blockShape({2, 2});
485  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {2, 0}};
486 
487  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
488  armnn::DataLayout::NCHW, inputShape, input, blockShape,
489  crops, outputShape, expectedOutput);
490 }

◆ BatchToSpaceNdNhwcTest1()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest1 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 89 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

93 {
94  const unsigned int inputShape[] = {4, 2, 2, 1};
95  const unsigned int outputShape[] = {1, 4, 4, 1};
96 
97  std::vector<float> input({
98  // Batch 0, Height 0, Width (2) x Channel (1)
99  1.0f, 3.0f,
100  // Batch 0, Height 1, Width (2) x Channel (1)
101  9.0f, 11.0f,
102 
103 
104  // Batch 1, Height 0, Width (2) x Channel (1)
105  2.0f, 4.0f,
106  // Batch 1, Height 1, Width (2) x Channel (1)
107  10.0f, 12.0f,
108 
109 
110  // Batch 2, Height 0, Width (2) x Channel (1)
111  5.0f, 7.0f,
112  // Batch 2, Height 1, Width (2) x Channel (1)
113  13.0f, 15.0f,
114 
115  // Batch 3, Height 0, Width (2) x Channel (3)
116  6.0f, 8.0f,
117  // Batch 3, Height 1, Width (2) x Channel (1)
118  14.0f, 16.0f
119  });
120 
121  std::vector<float> expectedOutput({
122  1.0f, 2.0f, 3.0f, 4.0f,
123  5.0f, 6.0f, 7.0f, 8.0f,
124  9.0f, 10.0f, 11.0f, 12.0f,
125  13.0f, 14.0f, 15.0f, 16.0f
126  });
127 
128  std::vector<unsigned int> blockShape {2, 2};
129  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
130 
131  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
132  armnn::DataLayout::NHWC, inputShape, input, blockShape,
133  crops, outputShape, expectedOutput);
134 }

◆ BatchToSpaceNdNhwcTest2()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest2 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 137 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

141 {
142  const unsigned int inputShape[] = {4, 1, 1, 1};
143  const unsigned int outputShape[] = {1, 2, 2, 1};
144 
145  std::vector<float> input({
146  // Batch 0, Height 0, Width (2) x Channel (1)
147  1.0f, 2.0f, 3.0f, 4.0f
148  });
149 
150  std::vector<float> expectedOutput({1.0f, 2.0f, 3.0f, 4.0f});
151 
152  std::vector<unsigned int> blockShape({2, 2});
153  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
154 
155  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
156  armnn::DataLayout::NHWC, inputShape, input, blockShape,
157  crops, outputShape, expectedOutput);
158 }

◆ BatchToSpaceNdNhwcTest3()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest3 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 161 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

165 {
166  const unsigned int inputShape[] = {4, 1, 1, 3};
167  const unsigned int outputShape[] = {1, 2, 2, 3};
168 
169  std::vector<float> input({1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f});
170 
171  std::vector<float> expectedOutput({1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f});
172 
173  std::vector<unsigned int> blockShape({2, 2});
174  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
175 
176  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
177  armnn::DataLayout::NHWC, inputShape, input, blockShape,
178  crops, outputShape, expectedOutput);
179 }

◆ BatchToSpaceNdNhwcTest4()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest4 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 182 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

186 {
187  const unsigned int inputShape[] = {8, 1, 3, 1};
188  const unsigned int outputShape[] = {2, 2, 4, 1};
189 
190  std::vector<float> input({
191  0.0f, 1.0f, 3.0f,
192  0.0f, 9.0f, 11.0f,
193  0.0f, 2.0f, 4.0f,
194  0.0f, 10.0f, 12.0f,
195  0.0f, 5.0f, 7.0f,
196  0.0f, 13.0f, 15.0f,
197  0.0f, 6.0f, 8.0f,
198  0.0f, 14.0f, 16.0f
199  });
200 
201  std::vector<float> expectedOutput({
202  1.0f, 2.0f, 3.0f, 4.0f,
203  5.0f, 6.0f, 7.0f, 8.0f,
204  9.0f, 10.0f, 11.0f, 12.0f,
205  13.0f, 14.0f, 15.0f, 16.0f
206  });
207 
208  std::vector<unsigned int> blockShape({2, 2});
209  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {2, 0}};
210 
211  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
212  armnn::DataLayout::NHWC, inputShape, input, blockShape,
213  crops, outputShape, expectedOutput);
214 }

◆ BatchToSpaceNdNhwcTest5()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest5 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 217 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

221 {
222  const unsigned int inputShape[] = {4, 2, 2, 1};
223  const unsigned int outputShape[] = {1, 4, 4, 1};
224 
225  std::vector<float> input({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16});
226  std::vector<float> expectedOutput({1, 5, 2, 6, 9, 13, 10, 14, 3, 7, 4, 8, 11, 15, 12, 16});
227 
228  std::vector<unsigned int> blockShape({2, 2});
229  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
230 
231  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
232  armnn::DataLayout::NHWC, inputShape,
233  input, blockShape, crops, outputShape, expectedOutput);
234 }

◆ BatchToSpaceNdNhwcTest6()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest6 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 237 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

241 {
242  const unsigned int inputShape[] = {4, 1, 1, 1};
243  const unsigned int outputShape[] = {1, 2, 2, 1};
244 
245  std::vector<float> input({
246  // Batch 0, Height 0, Width (2) x Channel (1)
247  1, 2, 3, 4
248  });
249 
250  std::vector<float> expectedOutput({1, 2, 3, 4});
251 
252  std::vector<unsigned int> blockShape({2, 2});
253  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
254 
255  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
256  armnn::DataLayout::NHWC, inputShape, input, blockShape,
257  crops, outputShape, expectedOutput);
258 }

◆ BatchToSpaceNdNhwcTest7()

LayerTestResult<T, 4> BatchToSpaceNdNhwcTest7 ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 261 of file BatchToSpaceNdTestImpl.hpp.

References armnn::NHWC.

265 {
266  const unsigned int inputShape[] = {4, 1, 1, 3};
267  const unsigned int outputShape[] = {1, 2, 2, 3};
268 
269  std::vector<float> input({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12});
270 
271  std::vector<float> expectedOutput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12});
272 
273  std::vector<unsigned int> blockShape({2, 2});
274  std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}};
275 
276  return BatchToSpaceNdHelper<ArmnnType, 4, 4>(workloadFactory, memoryManager, tensorHandleFactory,
277  armnn::DataLayout::NHWC, inputShape, input, blockShape,
278  crops, outputShape, expectedOutput);
279 }