| /external/armnn/src/backends/backendsCommon/test/ |
| D | SplitterEndToEndTestImpl.hpp | 24 const std::vector<TensorShape>& outputShapes, in CreateSplitterNetwork()
84 const std::vector<TensorShape> outputShapes{{ 2 }, { 2 }}; in Splitter1dEndToEnd() local
112 const std::vector<TensorShape> outputShapes{{ 2, 3 }, { 2, 3 }}; in Splitter2dDim0EndToEnd() local
147 const std::vector<TensorShape> outputShapes{{ 4, 1 }, { 4, 1 }, { 4, 1 }}; in Splitter2dDim1EndToEnd() local
185 const std::vector<TensorShape> outputShapes{{ 1, 4, 3 }, { 1, 4, 3 }}; in Splitter3dDim0EndToEnd() local
233 const std::vector<TensorShape> outputShapes{{ 2, 2, 3 }, { 2, 2, 3 }}; in Splitter3dDim1EndToEnd() local
281 const std::vector<TensorShape> outputShapes{{ 2, 4, 1 }, { 2, 4, 1 }, { 2, 4, 1 }}; in Splitter3dDim2EndToEnd() local
321 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim0EndToEnd() local
401 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim1EndToEnd() local
481 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim2EndToEnd() local
[all …]
|
| /external/armnn/src/armnn/test/ |
| D | InferOutputTests.hpp | 21 std::vector<armnn::TensorShape>& outputShapes) in ArgMinMaxInferOutputShapeImpl()
39 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape4dTest() local
58 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape3dTest() local
77 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape2dTest() local
96 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape1dTest() local
150 std::vector<armnn::TensorShape>& outputShapes) in PreluInferOutputShapeImpl()
170 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeSameDimsTest() local
190 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeInputBiggerTest() local
210 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeAlphaBiggerTest() local
230 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeNoMatchTest() local
[all …]
|
| /external/armnn/delegate/test/ |
| D | SplitTest.cpp | 25 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitUint8Test() local
55 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitFp32Test() local
140 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitVUint8Test() local
178 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitVFp32Test() local
|
| /external/android-nn-driver/ |
| D | ArmnnPreparedModel_1_3.cpp | 68 std::vector<V1_2::OutputShape> outputShapes, in NotifyCallbackAndCheck()
83 std::vector<V1_2::OutputShape> outputShapes, in NotifyCallbackAndCheck()
320 std::vector<V1_2::OutputShape> outputShapes, in execute_1_2()
345 std::vector<V1_2::OutputShape> outputShapes, in execute_1_3()
348 { in execute_1_3()
550 std::vector<V1_2::OutputShape> &outputShapes, in PrepareMemoryForOutputs()
646 std::vector<V1_2::OutputShape> outputShapes(request.outputs.size()); in PrepareMemoryForIO() local
741 { in executeSynchronously()
788 { in executeSynchronously_1_3()
838 std::vector<V1_2::OutputShape> outputShapes(outputTensors.size()); in ExecuteGraph() local
[all …]
|
| D | ArmnnPreparedModel_1_2.cpp | 63 std::vector<V1_2::OutputShape> outputShapes, in NotifyCallbackAndCheck()
295 std::vector<V1_2::OutputShape> outputShapes, in execute_1_2()
298 { in execute_1_2()
346 std::vector<V1_2::OutputShape> &outputShapes, in PrepareMemoryForOutputs()
430 std::vector<V1_2::OutputShape> outputShapes(request.outputs.size()); in PrepareMemoryForIO() local
493 { in executeSynchronously()
538 std::vector<V1_2::OutputShape> outputShapes(outputTensors.size()); in ExecuteGraph() local
776 std::vector<V1_2::OutputShape> outputShapes(outputTensorSize); in ScheduleGraphForExecution() local
|
| D | ArmnnPreparedModel_1_2.hpp | 101 std::vector<V1_2::OutputShape> outputShapes, in ArmnnThreadPoolCallback_1_2()
|
| D | ArmnnPreparedModel_1_3.hpp | 136 std::vector<V1_2::OutputShape> outputShapes, in ArmnnThreadPoolCallback_1_3()
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_ops.cpp | 27 std::vector<std::vector<ExprHandle>> outputShapes = {{N}, {M}, {}}; in TEST() local
57 std::vector<std::vector<ExprHandle>> outputShapes = { in TEST() local
|
| /external/armnn/shim/sl/canonical/ |
| D | ArmnnPreparedModel.cpp | 189 std::vector<OutputShape> &outputShapes, in PrepareMemoryForOutputs()
273 std::vector<OutputShape> outputShapes(request.outputs.size()); in PrepareMemoryForIO() local
353 std::vector<OutputShape> outputShapes(outputTensors->size()); in execute() local
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonTensorHandleTests.cpp | 259 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, variable
465 const std::vector<TensorShape> outputShapes{{ 1, 1, 2, 4 }, variable
|