Home
last modified time | relevance | path

Searched defs:outputTensorShape (Results 1 – 25 of 32) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/test/
H A DFullyConnectedTest.cpp16 std::vector<int32_t> outputTensorShape { 1, 1 }; in FullyConnectedFp32Test() local
42 std::vector<int32_t> outputTensorShape { 1, 1 }; in FullyConnectedActivationTest() local
68 std::vector<int32_t> outputTensorShape { 2, 1 }; in FullyConnectedInt8Test() local
H A DConvolutionTestHelper.hpp26 std::vector<char> CreateConv2dTfLiteModel(tflite::BuiltinOperator convolutionOperatorCode, in CreateConv2dTfLiteModel()
289 const std::vector<int32_t>& outputTensorShape, 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.hpp30 const std::vector<int32_t>& outputTensorShape) in CreateSliceTfLiteModel()
125 std::vector<int32_t>& outputTensorShape) in SliceTestImpl()
H A DStridedSliceTestHelper.hpp32 const std::vector<int32_t>& outputTensorShape, in CreateStridedSliceTfLiteModel()
149 std::vector<int32_t>& outputTensorShape, 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 DControlTestHelper.hpp24 std::vector<char> CreateConcatTfLiteModel(tflite::BuiltinOperator controlOperatorCode, in CreateConcatTfLiteModel()
115 std::vector<char> CreateMeanTfLiteModel(tflite::BuiltinOperator controlOperatorCode, in CreateMeanTfLiteModel()
H A DTransposeTestHelper.hpp23 std::vector<char> CreateTransposeTfLiteModel(tflite::TensorType tensorType, in CreateTransposeTfLiteModel()
H A DSpaceDepthTestHelper.hpp23 std::vector<char> CreateSpaceDepthTfLiteModel(tflite::BuiltinOperator spaceDepthOperatorCode, in CreateSpaceDepthTfLiteModel()
H A DShapeTestHelper.hpp26 const std::vector<int32_t>& outputTensorShape, in CreateShapeTfLiteModel()
H A DResizeTestHelper.hpp24 std::vector<char> CreateResizeTfLiteModel(tflite::BuiltinOperator operatorCode, in CreateResizeTfLiteModel()
H A DComparisonTestHelper.hpp24 std::vector<char> CreateComparisonTfLiteModel(tflite::BuiltinOperator comparisonOperatorCode, in CreateComparisonTfLiteModel()
H A DLogicalTestHelper.hpp24 std::vector<char> CreateLogicalBinaryTfLiteModel(tflite::BuiltinOperator logicalOperatorCode, in CreateLogicalBinaryTfLiteModel()
H A DQuantizationTestHelper.hpp24 std::vector<char> CreateQuantizationTfLiteModel(tflite::BuiltinOperator quantizationOperatorCode, in CreateQuantizationTfLiteModel()
H A DUnpackTestHelper.hpp24 std::vector<char> CreateUnpackTfLiteModel(tflite::BuiltinOperator unpackOperatorCode, in CreateUnpackTfLiteModel()
H A DPackTestHelper.hpp24 std::vector<char> CreatePackTfLiteModel(tflite::BuiltinOperator packOperatorCode, in CreatePackTfLiteModel()
H A DBatchMatMulTestHelper.hpp23 std::vector<char> CreateBatchMatMulTfLiteModel( in CreateBatchMatMulTfLiteModel()
H A DRedefineTestHelper.hpp28 const std::vector<int32_t>& outputTensorShape, in CreateRedefineTfLiteModel()
H A DArgMinMaxTestHelper.hpp29 const std::vector<int32_t>& outputTensorShape, in CreateArgMinMaxTfLiteModel()
H A DBatchSpaceTestHelper.hpp24 std::vector<char> CreateBatchSpaceTfLiteModel(tflite::BuiltinOperator batchSpaceOperatorCode, in CreateBatchSpaceTfLiteModel()
H A DReduceTestHelper.hpp24 std::vector<char> CreateReduceTfLiteModel(tflite::BuiltinOperator reduceOperatorCode, in CreateReduceTfLiteModel()
H A DElementwiseBinaryTestHelper.hpp25 std::vector<char> CreateElementwiseBinaryTfLiteModel(tflite::BuiltinOperator binaryOperatorCode, in CreateElementwiseBinaryTfLiteModel()
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DParserHelper.cpp93 … armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateReducedOutputTensoInfo() local
129 armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateStridedSliceOutputTensorInfo() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DReverse.cpp54 const Shape outputTensorShape = context->getOutputShape(kOutputTensor); in validate() local
DMirrorPad.cpp57 const Shape outputTensorShape = context->getOutputShape(kOutputTensor); in validate() local
/aosp_15_r20/external/armnn/src/armnn/test/
H A DInferOutputTests.hpp641 armnn::TensorShape outputTensorShape(2, outputShape.data()); in QLstmInferOutputShapeTest() local
689 armnn::TensorShape outputTensorShape(2, outputShape.data()); in QuantizedLstmInferOutputShapeTest() local

12