Home
last modified time | relevance | path

Searched defs:inputTensorShape (Results 1 – 25 of 43) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/test/
H A DSplitTestHelper.hpp26 std::vector<int32_t>& inputTensorShape, in CreateSplitTfLiteModel()
124 std::vector<int32_t>& inputTensorShape, in SplitTest()
173 std::vector<int32_t>& inputTensorShape, in CreateSplitVTfLiteModel()
281 std::vector<int32_t>& inputTensorShape, in SplitVTest()
H A DFullyConnectedTest.cpp13 std::vector<int32_t> inputTensorShape { 1, 4, 1, 1 }; in FullyConnectedFp32Test() local
39 std::vector<int32_t> inputTensorShape { 1, 4, 1, 1 }; in FullyConnectedActivationTest() local
65 std::vector<int32_t> inputTensorShape { 1, 4, 2, 1 }; in FullyConnectedInt8Test() local
H A DConvolutionTestHelper.hpp26 std::vector<char> CreateConv2dTfLiteModel(tflite::BuiltinOperator convolutionOperatorCode, in CreateConv2dTfLiteModel()
286 const std::vector<int32_t>& inputTensorShape, in CreateConv3dTfLiteModel()
515 std::vector<char> CreateTransposeConvTfLiteModel(tflite::TensorType tensorType, in CreateTransposeConvTfLiteModel()
635 void TransposeConvTest(std::vector<armnn::BackendId>& backends, in TransposeConvTest()
H A DSliceTestHelper.hpp25 const std::vector<int32_t>& inputTensorShape, in CreateSliceTfLiteModel()
122 std::vector<int32_t>& inputTensorShape, in SliceTestImpl()
H A DStridedSliceTestHelper.hpp25 const std::vector<int32_t>& inputTensorShape, in CreateStridedSliceTfLiteModel()
145 std::vector<int32_t>& inputTensorShape, in StridedSliceTestImpl()
H A DFullyConnectedTestHelper.hpp25 std::vector<char> CreateFullyConnectedTfLiteModel(tflite::TensorType tensorType, in CreateFullyConnectedTfLiteModel()
169 void FullyConnectedTest(std::vector<armnn::BackendId>& backends, in FullyConnectedTest()
H A DSpaceDepthTestHelper.hpp23 std::vector<char> CreateSpaceDepthTfLiteModel(tflite::BuiltinOperator spaceDepthOperatorCode, in CreateSpaceDepthTfLiteModel()
H A DShapeTestHelper.hpp25 const std::vector<int32_t>& inputTensorShape, in CreateShapeTfLiteModel()
H A DResizeTestHelper.hpp24 std::vector<char> CreateResizeTfLiteModel(tflite::BuiltinOperator operatorCode, in CreateResizeTfLiteModel()
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DParserPrototxtFixture.hpp97 …rPrototxtFixture<TParser>::SetupSingleInputSingleOutput(const armnn::TensorShape& inputTensorShape, in SetupSingleInputSingleOutput()
108 …rPrototxtFixture<TParser>::SetupSingleInputSingleOutput(const armnn::TensorShape& inputTensorShape, in SetupSingleInputSingleOutput()
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DReverse.cpp53 const Shape inputTensorShape = context->getInputShape(kInputTensor); in validate() local
DMirrorPad.cpp56 const Shape inputTensorShape = context->getInputShape(kInputTensor); in validate() local
/aosp_15_r20/external/armnn/tests/OnnxMnist-Armnn/
H A DOnnxMnist-Armnn.cpp11 armnn::TensorShape inputTensorShape({ 1, 1, 28, 28 }); in main() local
/aosp_15_r20/external/armnn/tests/OnnxMobileNet-Armnn/
H A DOnnxMobileNet-Armnn.cpp22 armnn::TensorShape inputTensorShape({ 1, 3, 224, 224 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteInceptionV4Quantized-Armnn/
H A DTfLiteInceptionV4Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteMobilenetV2Quantized-Armnn/
H A DTfLiteMobilenetV2Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteResNetV2-50-Quantized-Armnn/
H A DTfLiteResNetV2-50-Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteVGG16Quantized-Armnn/
H A DTfLiteVGG16Quantized-Armnn.cpp26 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteInceptionV3Quantized-Armnn/
H A DTfLiteInceptionV3Quantized-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteMobileNetQuantizedSoftmax-Armnn/
H A DTfLiteMobileNetQuantizedSoftmax-Armnn.cpp26 armnn::TensorShape inputTensorShape({ 1, 128, 128, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteMnasNet-Armnn/
H A DTfLiteMnasNet-Armnn.cpp24 armnn::TensorShape inputTensorShape({ 2, 224, 224, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteResNetV2-Armnn/
H A DTfLiteResNetV2-Armnn.cpp25 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); in main() local
/aosp_15_r20/external/armnn/tests/TfLiteMobileNetSsd-Armnn/
H A DTfLiteMobileNetSsd-Armnn.cpp23 armnn::TensorShape inputTensorShape({ 1, 300, 300, 3 }); in main() local
/aosp_15_r20/external/armnn/src/armnn/test/
H A DInferOutputTests.hpp624 armnn::TensorShape inputTensorShape(2, inputShape.data()); in QLstmInferOutputShapeTest() local
674 armnn::TensorShape inputTensorShape(2, inputShape.data()); in QuantizedLstmInferOutputShapeTest() local
/aosp_15_r20/external/armnn/tests/TfLiteMobilenetQuantized-Armnn/
H A DTfLiteMobilenetQuantized-Armnn.cpp105 armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); in main() local

12