Home
last modified time | relevance | path

Searched refs:inputTensorShape (Results 1 – 25 of 42) sorted by relevance

12

/external/armnn/delegate/test/
DFullyConnectedTest.cpp13 std::vector<int32_t> inputTensorShape { 1, 4, 1, 1 }; in FullyConnectedFp32Test() local
27 inputTensorShape, in FullyConnectedFp32Test()
39 std::vector<int32_t> inputTensorShape { 1, 4, 1, 1 }; in FullyConnectedActivationTest() local
53 inputTensorShape, in FullyConnectedActivationTest()
65 std::vector<int32_t> inputTensorShape { 1, 4, 2, 1 }; in FullyConnectedInt8Test() local
81 inputTensorShape, in FullyConnectedInt8Test()
DSplitTestHelper.hpp26 std::vector<int32_t>& inputTensorShape, in CreateSplitTfLiteModel() argument
59 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateSplitTfLiteModel()
60 inputTensorShape.size()), in CreateSplitTfLiteModel()
124 std::vector<int32_t>& inputTensorShape, in SplitTest() argument
136 inputTensorShape, in SplitTest()
173 std::vector<int32_t>& inputTensorShape, in CreateSplitVTfLiteModel() argument
204 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateSplitVTfLiteModel()
205 inputTensorShape.size()), in CreateSplitVTfLiteModel()
281 std::vector<int32_t>& inputTensorShape, in SplitVTest() argument
295 inputTensorShape, in SplitVTest()
DSliceTestHelper.hpp25 const std::vector<int32_t>& inputTensorShape, in CreateSliceTfLiteModel() argument
49 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateSliceTfLiteModel()
50 inputTensorShape.size()), in CreateSliceTfLiteModel()
122 std::vector<int32_t>& inputTensorShape, in SliceTestImpl() argument
130 inputTensorShape, in SliceTestImpl()
DStridedSliceTestHelper.hpp25 const std::vector<int32_t>& inputTensorShape, in CreateStridedSliceTfLiteModel() argument
60 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateStridedSliceTfLiteModel()
61 inputTensorShape.size()), in CreateStridedSliceTfLiteModel()
145 std::vector<int32_t>& inputTensorShape, in StridedSliceTestImpl() argument
160 inputTensorShape, in StridedSliceTestImpl()
DFullyConnectedTestHelper.hpp27 const std::vector <int32_t>& inputTensorShape, in CreateFullyConnectedTfLiteModel()
94 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateFullyConnectedTfLiteModel()
95 inputTensorShape.size()), in CreateFullyConnectedTfLiteModel()
172 const std::vector <int32_t>& inputTensorShape, in FullyConnectedTest()
187 inputTensorShape, in FullyConnectedTest()
DConvolutionTestHelper.hpp34 const std::vector <int32_t>& inputTensorShape, in CreateConv2dTfLiteModel()
98 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(),
99 inputTensorShape.size()),
286 const std::vector<int32_t>& inputTensorShape, in CreateConv3dTfLiteModel() argument
348 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(),
349 inputTensorShape.size()),
521 const std::vector <int32_t>& inputTensorShape, in CreateTransposeConvTfLiteModel()
577 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateTransposeConvTfLiteModel()
578 inputTensorShape.size()), in CreateTransposeConvTfLiteModel()
642 const std::vector <int32_t>& inputTensorShape, in TransposeConvTest()
[all …]
DShapeTestHelper.hpp25 const std::vector<int32_t>& inputTensorShape, in CreateShapeTfLiteModel() argument
47 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateShapeTfLiteModel()
48 inputTensorShape.size()), in CreateShapeTfLiteModel()
DQuantizationTestHelper.hpp27 const std::vector <int32_t>& inputTensorShape, in CreateQuantizationTfLiteModel()
51 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateQuantizationTfLiteModel()
52 inputTensorShape.size()), in CreateQuantizationTfLiteModel()
DUnpackTestHelper.hpp26 std::vector<int32_t>& inputTensorShape, in CreateUnpackTfLiteModel() argument
57 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateUnpackTfLiteModel()
58 inputTensorShape.size()), in CreateUnpackTfLiteModel()
DSpaceDepthTestHelper.hpp25 const std::vector <int32_t>& inputTensorShape, in CreateSpaceDepthTfLiteModel()
46 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreateSpaceDepthTfLiteModel()
47 inputTensorShape.size()), in CreateSpaceDepthTfLiteModel()
DPackTestHelper.hpp26 std::vector<int32_t>& inputTensorShape, in CreatePackTfLiteModel() argument
56 flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), in CreatePackTfLiteModel()
57 inputTensorShape.size()), in CreatePackTfLiteModel()
DResizeTestHelper.hpp26 const std::vector <int32_t>& inputTensorShape, in CreateResizeTfLiteModel()
45 … flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), inputTensorShape.size()), in CreateResizeTfLiteModel()
/external/armnn/tests/TfLiteMobileNetSsd-Armnn/
DTfLiteMobileNetSsd-Armnn.cpp23 armnn::TensorShape inputTensorShape({ 1, 300, 300, 3 }); in main() local
39 [&inputTensorShape, inputLayerNames, outputLayerNames]() in main()
61 modelParams.m_InputShapes = { inputTensorShape }; in main()
/external/armnn/tests/OnnxMnist-Armnn/
DOnnxMnist-Armnn.cpp11 armnn::TensorShape inputTensorShape({ 1, 1, 28, 28 }); in main() local
28 &inputTensorShape); in main()
/external/armnn/tests/TfLiteResNetV2-Armnn/
DTfLiteResNetV2-Armnn.cpp25 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); in main() local
49 &inputTensorShape); in main()
/external/armnn/tests/TfLiteInceptionV3Quantized-Armnn/
DTfLiteInceptionV3Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); in main() local
50 &inputTensorShape); in main()
/external/armnn/tests/TfLiteResNetV2-50-Quantized-Armnn/
DTfLiteResNetV2-50-Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
50 &inputTensorShape); in main()
/external/armnn/tests/OnnxMobileNet-Armnn/
DOnnxMobileNet-Armnn.cpp22 armnn::TensorShape inputTensorShape({ 1, 3, 224, 224 }); in main() local
48 &inputTensorShape); in main()
/external/armnn/tests/TfLiteMobilenetV2Quantized-Armnn/
DTfLiteMobilenetV2Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
50 &inputTensorShape); in main()
/external/armnn/tests/TfLiteInceptionV4Quantized-Armnn/
DTfLiteInceptionV4Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); in main() local
50 &inputTensorShape); in main()
/external/armnn/tests/TfLiteMnasNet-Armnn/
DTfLiteMnasNet-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 2, 224, 224, 3 }); in main() local
49 &inputTensorShape); in main()
/external/armnn/tests/TfLiteMobileNetQuantizedSoftmax-Armnn/
DTfLiteMobileNetQuantizedSoftmax-Armnn.cpp26 armnn::TensorShape inputTensorShape({ 1, 128, 128, 3 }); in main() local
56 &inputTensorShape); in main()
/external/armnn/tests/TfLiteVGG16Quantized-Armnn/
DTfLiteVGG16Quantized-Armnn.cpp26 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
56 &inputTensorShape); in main()
/external/armnn/src/armnnUtils/
DParserPrototxtFixture.hpp36 void SetupSingleInputSingleOutput(const armnn::TensorShape& inputTensorShape,
39 void SetupSingleInputSingleOutput(const armnn::TensorShape& inputTensorShape,
97 …rPrototxtFixture<TParser>::SetupSingleInputSingleOutput(const armnn::TensorShape& inputTensorShape, in SetupSingleInputSingleOutput() argument
104 Setup({ { inputName, inputTensorShape } }, { outputName }); in SetupSingleInputSingleOutput()
108 …rPrototxtFixture<TParser>::SetupSingleInputSingleOutput(const armnn::TensorShape& inputTensorShape, in SetupSingleInputSingleOutput() argument
118 Setup({ { inputName, inputTensorShape } }, { outputName }); in SetupSingleInputSingleOutput()
/external/armnn/tests/TfLiteMobilenetQuantized-Armnn/
DTfLiteMobilenetQuantized-Armnn.cpp105 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
131 &inputTensorShape); in main()

12