| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | BatchToSpaceNdTestImpl.hpp | 33 const std::vector<unsigned int> &blockShape, in BatchToSpaceNdHelper()
128 std::vector<unsigned int> blockShape {2, 2}; in BatchToSpaceNdNhwcTest1() local
152 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNhwcTest2() local
173 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNhwcTest3() local
208 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNhwcTest4() local
228 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNhwcTest5() local
252 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNhwcTest6() local
273 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNhwcTest7() local
306 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNchwTest1() local
330 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNdNchwTest2() local
[all …]
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | BatchSpaceTest.cpp | 27 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNDFp32Test() local
49 std::vector<unsigned int> blockShape({1, 1}); in BatchToSpaceNDFp32BatchOneTest() local
71 std::vector<unsigned int> blockShape({2, 2}); in BatchToSpaceNDUint8Test() local
94 std::vector<unsigned int> blockShape({2, 2}); in SpaceToBatchNDFp32Test() local
123 std::vector<unsigned int> blockShape({2, 2}); in SpaceToBatchNDFp32PaddingTest() local
145 std::vector<unsigned int> blockShape({2, 2}); in SpaceToBatchNDUint8Test() local
|
| H A D | BatchSpaceTestHelper.hpp | 57 std::vector<int32_t> blockShape { 2 }; in CreateBatchSpaceTfLiteModel() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | BatchToSpaceNdEndToEndTestImpl.hpp | 23 std::vector<unsigned int>& blockShape, in CreateBatchToSpaceNdNetwork()
55 std::vector<unsigned int> blockShape {2, 2}; in BatchToSpaceNdEndToEnd() local
84 std::vector<unsigned int> blockShape {2, 2}; in BatchToSpaceNdComplexEndToEnd() local
|
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | BatchSpace.hpp | 55 std::vector<unsigned int> blockShape(blockShapeTensorInfo.GetNumElements()); in VisitBatchToSpaceNdOperator() local
155 std::vector<unsigned int> blockShape(blockShapeTensorInfo.GetNumElements()); in VisitSpaceToBatchNdOperator() local
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | ShapeInferenceTests.cpp | 206 std::vector<unsigned int> blockShape {2, 2}; variable
654 std::vector<unsigned int> blockShape {2, 2}; variable
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | BatchToSpaceNd.cpp | 38 const std::vector<unsigned int>& blockShape, in BatchToSpaceNd()
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 1758 std::vector<unsigned int> blockShape(blockShapeTensorInfo.GetNumElements()); in ParseBatchToSpaceND() local
2107 std::vector<unsigned int> blockShape(blockShapeTensorInfo.GetNumElements()); in ParseSpaceToBatchND() local
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ConversionUtils.hpp | 4144 std::vector<int32_t> blockShape; in ConvertSpaceToBatchNd() local
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | Converter.cpp | 5092 std::vector<int32_t> blockShape; in ConvertSpaceToBatchNd() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2020 ANeuralNetworksOperandType blockShape = {.type = ANEURALNETWORKS_TENSOR_INT32, in spaceBatchOpTest() local
|