Searched refs:inputTensorShapes (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/tosaCommon/test/ |
| H A D | OneToOneMappingTests.cpp | 63 std::vector<armnn::TensorShape> inputTensorShapes = { { 2, 3, 2, 2 }, { 2, 3, 2, 2 } }; variable
64 armnn::TensorInfo input0Info(inputTensorShapes[0], DataType::Float32);
65 armnn::TensorInfo input1Info(inputTensorShapes[1], DataType::Float32);
71 descriptor = armnn::CreateDescriptorForConcatenation(inputTensorShapes.begin(),
72 inputTensorShapes.end(),
101 std::vector<armnn::TensorShape> inputTensorShapes = { { 2, 3, 2, 2 }, { 2, 3, 2, 2 } }; variable
102 descriptor = armnn::CreateDescriptorForConcatenation(inputTensorShapes.begin(),
103 inputTensorShapes.end(),
116 TensorInfo inputInfo0 = TensorInfo(inputTensorShapes[0], DataType::Float32, 0.0f, 0, true);
117 TensorInfo inputInfo1 = TensorInfo(inputTensorShapes[1], DataType::Float32, 0.0f, 0, true);
|
| /aosp_15_r20/external/armnn/src/armnnConverter/ |
| H A D | ArmnnConverter.cpp | 355 std::vector<armnn::TensorShape> inputTensorShapes; in main() local
378 inputTensorShapes.push_back(shape); in main()
388 …ArmnnConverter converter(modelPath, inputNames, inputTensorShapes, outputNames, outputPath, isMode… in main()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Pack.cpp | 116 const tflite::RuntimeShape inputTensorShapes = in pack() local
119 &inputTensorShapes); in pack()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | ConcatTestImpl.cpp | 2095 …std::vector<TensorShape> inputTensorShapes({inputTensorInfo1.GetShape(), inputTensorInfo2.GetShape… in ConcatDifferentInputOutputQParamTest() local
2191 inputTensorShapes.begin(),inputTensorShapes.end(), 2); in ConcatDifferentInputOutputQParamTest()
|