20 template<
typename T, std::
size_t InDim, std::
size_t OutDim>
27 std::vector<float>& inputData,
28 std::vector<float>& outputExpectedData,
30 const float qScale = 1.0f,
31 const int32_t qOffset = 0)
34 if(armnn::IsQuantizedType<T>())
43 std::vector<T> input = armnnUtils::QuantizedVector<T>(inputData, qScale, qOffset);
44 std::vector<T> expectedOutput = armnnUtils::QuantizedVector<T>(outputExpectedData, qScale, qOffset);
47 std::unique_ptr<armnn::ITensorHandle> inputHandle =
50 std::unique_ptr<armnn::ITensorHandle> outputHandle =
54 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get());
55 AddOutputToWorkload(descriptor, info, outputTensorInfo, outputHandle.get());
61 inputHandle->Allocate();
62 outputHandle->Allocate();
66 ExecuteWorkload(*workload, memoryManager);
72 outputHandle->GetShape(),
76 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
85 unsigned int inputShape[] = {3, 2, 3, 1};
86 unsigned int outputShape[] = {1, 2, 3, 1};
96 std::vector<float> input = std::vector<float>(
98 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
100 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
102 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
105 std::vector<float> outputExpected = std::vector<float>(
107 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f
110 return StridedSliceTestImpl<T, 4, 4>(
111 workloadFactory, memoryManager, tensorHandleFactory,
112 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
115 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
124 unsigned int inputShape[] = {3, 2, 3, 1};
125 unsigned int outputShape[] = {1, 2, 3, 1};
135 std::vector<float> input = std::vector<float>(
137 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
139 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
141 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
144 std::vector<float> outputExpected = std::vector<float>(
146 4.0f, 4.0f, 4.0f, 3.0f, 3.0f, 3.0f
149 return StridedSliceTestImpl<T, 4, 4>(
150 workloadFactory, memoryManager, tensorHandleFactory,
151 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
154 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
163 unsigned int inputShape[] = {3, 2, 3, 1};
164 unsigned int outputShape[] = {2, 1, 2, 1};
174 std::vector<float> input = std::vector<float>(
176 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
178 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
180 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
183 std::vector<float> outputExpected = std::vector<float>(
190 return StridedSliceTestImpl<T, 4, 4>(
191 workloadFactory, memoryManager, tensorHandleFactory,
192 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
195 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
204 unsigned int inputShape[] = {3, 2, 3, 1};
205 unsigned int outputShape[] = {3, 2, 3, 1};
217 std::vector<float> input = std::vector<float>(
219 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
221 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
223 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
226 std::vector<float> outputExpected = std::vector<float>(
228 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
230 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
232 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
235 return StridedSliceTestImpl<T, 4, 4>(
236 workloadFactory, memoryManager, tensorHandleFactory,
237 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
240 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
249 unsigned int inputShape[] = {3, 2, 3, 1};
250 unsigned int outputShape[] = {3, 1};
262 std::vector<float> input = std::vector<float>(
264 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
266 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
268 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
271 std::vector<float> outputExpected = std::vector<float>(
276 return StridedSliceTestImpl<T, 4, 2>(
277 workloadFactory, memoryManager, tensorHandleFactory,
278 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
281 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
290 unsigned int inputShape[] = {3, 2, 3, 1};
291 unsigned int outputShape[] = {2, 3, 1};
303 std::vector<float> input = std::vector<float>(
305 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
307 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
309 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
312 std::vector<float> outputExpected = std::vector<float>(
314 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f
317 return StridedSliceTestImpl<T, 4, 3>(
318 workloadFactory, memoryManager, tensorHandleFactory,
319 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
322 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
331 unsigned int inputShape[] = {3, 2, 3, 1};
332 unsigned int outputShape[] = {3, 3, 1};
344 std::vector<float> input = std::vector<float>(
346 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
348 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
350 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
353 std::vector<float> outputExpected = std::vector<float>(
355 1.0f, 2.0f, 3.0f, 7.0f, 8.0f, 9.0f, 13.0f, 14.0f, 15.0f
358 return StridedSliceTestImpl<T, 4, 3>(
359 workloadFactory, memoryManager, tensorHandleFactory,
360 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
363 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
372 unsigned int inputShape[] = {3, 2, 3, 1};
373 unsigned int outputShape[] = {3, 2, 1};
385 std::vector<float> input = std::vector<float>(
387 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
389 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
391 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
394 std::vector<float> outputExpected = std::vector<float>(
396 1.0f, 4.0f, 7.0f, 10.0f, 13.0f, 16.0f
399 return StridedSliceTestImpl<T, 4, 3>(
400 workloadFactory, memoryManager, tensorHandleFactory,
401 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
404 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
413 unsigned int inputShape[] = {3, 2, 3, 1};
414 unsigned int outputShape[] = {3, 2, 3};
426 std::vector<float> input = std::vector<float>(
428 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
430 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
432 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
435 std::vector<float> outputExpected = std::vector<float>(
437 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
439 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
441 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
444 return StridedSliceTestImpl<T, 4, 3>(
445 workloadFactory, memoryManager, tensorHandleFactory,
446 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
449 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
458 unsigned int inputShape[] = {3, 2, 3, 1};
459 unsigned int outputShape[] = {3, 1};
471 std::vector<float> input = std::vector<float>(
473 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
475 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
477 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
480 std::vector<float> outputExpected = std::vector<float>(
485 return StridedSliceTestImpl<T, 4, 2>(
486 workloadFactory, memoryManager, tensorHandleFactory,
487 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
490 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
499 unsigned int inputShape[] = {2, 3, 1};
500 unsigned int outputShape[] = {3, 1};
512 std::vector<float> input = std::vector<float>(
514 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f
517 std::vector<float> outputExpected = std::vector<float>(
522 return StridedSliceTestImpl<T, 3, 2>(
523 workloadFactory, memoryManager, tensorHandleFactory,
524 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
527 void FillVector(std::vector<float>& inputArray,
float start,
float step)
529 for (uint32_t i = 0; i < inputArray.size(); ++i)
531 inputArray[i] = start;
536 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
545 unsigned int inputShape[] = {1, 1, 8, 942};
546 unsigned int outputShape[] = {1, 1, 1, 279};
560 std::vector<float> input(7536);
561 FillVector(input, 1.0f, 1.0f);
564 std::vector<float> outputExpected(279);
565 FillVector(outputExpected, 1171.0, 2.0f);
567 return StridedSliceTestImpl<T, 4, 4>(
568 workloadFactory, memoryManager, tensorHandleFactory,
569 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
572 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
581 unsigned int inputShape[] = {3, 2, 3, 1};
582 unsigned int outputShape[] = {2, 1};
594 std::vector<float> input = std::vector<float>(
596 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
598 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
600 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
603 std::vector<float> outputExpected = std::vector<float>(
608 return StridedSliceTestImpl<T, 4, 2>(
609 workloadFactory, memoryManager, tensorHandleFactory,
610 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
613 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
622 unsigned int inputShape[] = {3, 2, 3, 1};
623 unsigned int outputShape[] = {2, 3};
635 std::vector<float> input = std::vector<float>(
637 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
639 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
641 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
644 std::vector<float> outputExpected = std::vector<float>(
646 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f
649 return StridedSliceTestImpl<T, 4, 2>(
650 workloadFactory, memoryManager, tensorHandleFactory,
651 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
654 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
663 unsigned int inputShape[] = {3, 2, 3, 1};
664 unsigned int outputShape[] = {3};
676 std::vector<float> input = std::vector<float>(
678 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
680 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
682 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f
685 std::vector<float> outputExpected = std::vector<float>(
690 return StridedSliceTestImpl<T, 4, 1>(
691 workloadFactory, memoryManager, tensorHandleFactory,
692 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
695 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
704 unsigned int inputShape[] = {3, 3, 3};
705 unsigned int outputShape[] = {2, 2, 2};
716 std::vector<float> input = std::vector<float>(
718 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f,
720 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f,
722 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f, 25.0f, 26.0f, 27.0f
725 std::vector<float> outputExpected = std::vector<float>(
727 1.0f, 3.0f, 7.0f, 9.0f,
729 19.0f, 21.0f, 25.0f, 27.0f
732 return StridedSliceTestImpl<T, 3, 3>(
733 workloadFactory, memoryManager, tensorHandleFactory,
734 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
737 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
746 unsigned int inputShape[] = {3, 3, 3};
747 unsigned int outputShape[] = {2, 2, 2};
757 std::vector<float> input = std::vector<float>(
759 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f,
761 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f,
763 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f, 25.0f, 26.0f, 27.0f
766 std::vector<float> outputExpected = std::vector<float>(
768 27.0f, 25.0f, 21.0f, 19.0f,
770 9.0f, 7.0f, 3.0f, 1.0f
773 return StridedSliceTestImpl<T, 3, 3>(
774 workloadFactory, memoryManager, tensorHandleFactory,
775 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
778 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
787 unsigned int inputShape[] = {3, 3};
788 unsigned int outputShape[] = {2, 2};
799 std::vector<float> input = std::vector<float>(
808 std::vector<float> outputExpected = std::vector<float>(
815 return StridedSliceTestImpl<T, 2, 2>(
816 workloadFactory, memoryManager, tensorHandleFactory,
817 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
820 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
829 unsigned int inputShape[] = {3, 3};
830 unsigned int outputShape[] = {2, 2};
842 std::vector<float> input = std::vector<float>(
851 std::vector<float> outputExpected = std::vector<float>(
858 return StridedSliceTestImpl<T, 2, 2>(
859 workloadFactory, memoryManager, tensorHandleFactory,
860 inputTensorInfo, outputTensorInfo, input, outputExpected, desc);
870 return StridedSlice4dTest<armnn::DataType::Float32>(workloadFactory,
872 tensorHandleFactory);
880 return StridedSlice4dReverseTest<armnn::DataType::Float32>(workloadFactory,
882 tensorHandleFactory);
890 return StridedSliceSimpleStrideTest<armnn::DataType::Float32>(workloadFactory,
892 tensorHandleFactory);
900 return StridedSliceSimpleRangeMaskTest<armnn::DataType::Float32>(workloadFactory,
902 tensorHandleFactory);
910 return StridedSliceShrinkAxisMaskTest<armnn::DataType::Float32>(workloadFactory,
912 tensorHandleFactory);
920 return StridedSliceShrinkAxisMaskCTSTest<armnn::DataType::Float32>(workloadFactory,
922 tensorHandleFactory);
930 return StridedSliceShrinkAxisMaskBitPosition0Dim3Test<armnn::DataType::Float32>(workloadFactory,
932 tensorHandleFactory);
940 return StridedSliceShrinkAxisMaskBitPosition0Test<armnn::DataType::Float32>(workloadFactory,
942 tensorHandleFactory);
950 return StridedSliceShrinkAxisMaskBitPosition1Test<armnn::DataType::Float32>(workloadFactory,
952 tensorHandleFactory);
960 return StridedSliceShrinkAxisMaskBitPosition2Test<armnn::DataType::Float32>(workloadFactory,
962 tensorHandleFactory);
970 return StridedSliceShrinkAxisMaskBitPosition3Test<armnn::DataType::Float32>(workloadFactory,
972 tensorHandleFactory);
980 return StridedSliceShrinkAxisMaskBitPosition0And1Test<armnn::DataType::Float32>(workloadFactory,
982 tensorHandleFactory);
990 return StridedSliceShrinkAxisMaskBitPosition0And2Test<armnn::DataType::Float32>(workloadFactory,
992 tensorHandleFactory);
1000 return StridedSliceShrinkAxisMaskBitPosition0And3Test<armnn::DataType::Float32>(workloadFactory,
1002 tensorHandleFactory);
1010 return StridedSliceShrinkAxisMaskBitPosition0And1And3Test<armnn::DataType::Float32>(workloadFactory,
1012 tensorHandleFactory);
1020 return StridedSlice3dTest<armnn::DataType::Float32>(workloadFactory,
1022 tensorHandleFactory);
1030 return StridedSlice3dReverseTest<armnn::DataType::Float32>(workloadFactory,
1032 tensorHandleFactory);
1040 return StridedSlice2dTest<armnn::DataType::Float32>(workloadFactory,
1042 tensorHandleFactory);
1050 return StridedSlice2dReverseTest<armnn::DataType::Float32>(workloadFactory,
1052 tensorHandleFactory);
1060 return StridedSlice4dTest<armnn::DataType::QAsymmU8>(workloadFactory,
1062 tensorHandleFactory);
1070 return StridedSlice4dReverseTest<armnn::DataType::QAsymmU8>(workloadFactory,
1072 tensorHandleFactory);
1080 return StridedSliceSimpleStrideTest<armnn::DataType::QAsymmU8>(workloadFactory,
1082 tensorHandleFactory);
1090 return StridedSliceSimpleRangeMaskTest<armnn::DataType::QAsymmU8>(workloadFactory,
1092 tensorHandleFactory);
1100 return StridedSliceShrinkAxisMaskTest<armnn::DataType::QAsymmU8>(workloadFactory,
1102 tensorHandleFactory);
1110 return StridedSliceShrinkAxisMaskBitPosition0Dim3Test<armnn::DataType::QAsymmU8>(workloadFactory,
1112 tensorHandleFactory);
1120 return StridedSliceShrinkAxisMaskBitPosition0Test<armnn::DataType::QAsymmU8>(workloadFactory,
1122 tensorHandleFactory);
1130 return StridedSliceShrinkAxisMaskBitPosition1Test<armnn::DataType::QAsymmU8>(workloadFactory,
1132 tensorHandleFactory);
1140 return StridedSliceShrinkAxisMaskBitPosition2Test<armnn::DataType::QAsymmU8>(workloadFactory,
1142 tensorHandleFactory);
1150 return StridedSliceShrinkAxisMaskBitPosition3Test<armnn::DataType::QAsymmU8>(workloadFactory,
1152 tensorHandleFactory);
1160 return StridedSliceShrinkAxisMaskBitPosition0And1Test<armnn::DataType::QAsymmU8>(workloadFactory,
1162 tensorHandleFactory);
1170 return StridedSliceShrinkAxisMaskBitPosition0And2Test<armnn::DataType::QAsymmU8>(workloadFactory,
1172 tensorHandleFactory);
1180 return StridedSliceShrinkAxisMaskBitPosition0And3Test<armnn::DataType::QAsymmU8>(workloadFactory,
1182 tensorHandleFactory);
1190 return StridedSliceShrinkAxisMaskBitPosition0And1And3Test<armnn::DataType::QAsymmU8>(workloadFactory,
1192 tensorHandleFactory);
1200 return StridedSlice3dTest<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory);
1208 return StridedSlice3dReverseTest<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory);
1216 return StridedSlice2dTest<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory);
1224 return StridedSlice2dReverseTest<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory);
1232 return StridedSlice4dTest<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory);
1240 return StridedSlice4dReverseTest<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory);
1248 return StridedSliceSimpleStrideTest<armnn::DataType::QSymmS16>(workloadFactory,
1250 tensorHandleFactory);
1258 return StridedSliceSimpleRangeMaskTest<armnn::DataType::QSymmS16>(workloadFactory,
1260 tensorHandleFactory);
1268 return StridedSliceShrinkAxisMaskTest<armnn::DataType::QSymmS16>(workloadFactory,
1270 tensorHandleFactory);
1278 return StridedSlice3dTest<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory);
1286 return StridedSlice3dReverseTest<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory);
1294 return StridedSlice2dTest<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory);
1302 return StridedSlice2dReverseTest<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory);
LayerTestResult< uint8_t, 3 > StridedSlice3dUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 4 > StridedSliceShrinkAxisMaskCTSFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 3 > StridedSliceShrinkAxisMaskBitPosition2Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSlice2dFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 3 > StridedSliceShrinkAxisMaskBitPosition3Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
const TensorShape & GetShape() const
LayerTestResult< float, 3 > StridedSliceShrinkAxisMaskBitPosition3Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
int32_t m_ShrinkAxisMask
Shrink axis mask value. If set, the nth specification shrinks the dimensionality by 1...
std::vector< int > m_Begin
Begin values for the input that will be sliced.
LayerTestResult< uint8_t, 2 > StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSlice2dReverseFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 4 > StridedSliceSimpleRangeMaskFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 3 > StridedSliceShrinkAxisMaskBitPosition2Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSliceShrinkAxisMaskFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 2 > StridedSlice2dReverseUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 2 > StridedSliceShrinkAxisMaskInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
void IgnoreUnused(Ts &&...)
int32_t m_BeginMask
Begin mask value.
LayerTestResult< float, 3 > StridedSlice3dFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
int32_t m_EndMask
End mask value.
LayerTestResult< float, 1 > StridedSliceShrinkAxisMaskBitPosition0And1And3Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 2 > StridedSliceShrinkAxisMaskBitPosition0And3Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerDescriptor m_Parameters
LayerTestResult< float, 3 > StridedSliceShrinkAxisMaskBitPosition0Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 4 > StridedSlice4dInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 4 > StridedSlice4dReverseInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 2 > StridedSlice2dUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSliceShrinkAxisMaskBitPosition0And3Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 3 > StridedSlice3dReverseInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSliceShrinkAxisMaskBitPosition0And1Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
LayerTestResult< uint8_t, 3 > StridedSliceShrinkAxisMaskBitPosition0Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 4 > StridedSlice4dFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 1 > StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
void CopyDataFromITensorHandle(void *mem, const armnn::ITensorHandle *tensorHandle)
void SetQuantizationScale(float scale)
LayerTestResult< uint8_t, 3 > StridedSliceShrinkAxisMaskBitPosition1Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 3 > StridedSlice3dReverseUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
std::vector< int > m_Stride
Stride values for the input that will be sliced.
LayerTestResult< int16_t, 2 > StridedSlice2dReverseInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSliceShrinkAxisMaskBitPosition0Dim3Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 3 > StridedSliceShrinkAxisMaskBitPosition1Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 4 > StridedSliceSimpleStrideInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 3 > StridedSlice3dReverseFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 4 > StridedSliceSimpleRangeMaskInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
std::vector< int > m_End
End values for the input that will be sliced.
LayerTestResult< uint8_t, 4 > StridedSlice4dReverseUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 2 > StridedSliceShrinkAxisMaskBitPosition0And2Float32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 3 > StridedSlice3dInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
void CopyDataToITensorHandle(armnn::ITensorHandle *tensorHandle, const void *memory)
LayerTestResult< uint8_t, 4 > StridedSlice4dUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 2 > StridedSliceShrinkAxisMaskBitPosition0And2Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< int16_t, 2 > StridedSlice2dInt16Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
Contains information about TensorInfos of a layer.
LayerTestResult< uint8_t, 4 > StridedSliceSimpleStrideUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
void SetQuantizationOffset(int32_t offset)
LayerTestResult< float, 4 > StridedSliceSimpleStrideFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< uint8_t, 4 > StridedSliceSimpleRangeMaskUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< float, 4 > StridedSlice4dReverseFloat32Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
virtual std::unique_ptr< IWorkload > CreateWorkload(LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const
virtual std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo) const =0
LayerTestResult< uint8_t, 2 > StridedSliceShrinkAxisMaskBitPosition0And1Uint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
unsigned int GetNumElements() const
LayerTestResult< uint8_t, 2 > StridedSliceShrinkAxisMaskUint8Test(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)