Home
last modified time | relevance | path

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

12

/external/armnn/delegate/test/
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
DSliceTestHelper.hpp30 const std::vector<int32_t>& outputTensorShape) in CreateSliceTfLiteModel()
125 std::vector<int32_t>& outputTensorShape) in SliceTestImpl()
DStridedSliceTestHelper.hpp32 const std::vector<int32_t>& outputTensorShape, in CreateStridedSliceTfLiteModel()
149 std::vector<int32_t>& outputTensorShape, in StridedSliceTestImpl()
DFullyConnectedTestHelper.hpp25 std::vector<char> CreateFullyConnectedTfLiteModel(tflite::TensorType tensorType, in CreateFullyConnectedTfLiteModel()
169 void FullyConnectedTest(std::vector<armnn::BackendId>& backends, in FullyConnectedTest()
DControlTestHelper.hpp24 std::vector<char> CreateConcatTfLiteModel(tflite::BuiltinOperator controlOperatorCode, in CreateConcatTfLiteModel()
115 std::vector<char> CreateMeanTfLiteModel(tflite::BuiltinOperator controlOperatorCode, in CreateMeanTfLiteModel()
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()
DTransposeTestHelper.hpp23 std::vector<char> CreateTransposeTfLiteModel(tflite::TensorType tensorType, in CreateTransposeTfLiteModel()
DSpaceDepthTestHelper.hpp23 std::vector<char> CreateSpaceDepthTfLiteModel(tflite::BuiltinOperator spaceDepthOperatorCode, in CreateSpaceDepthTfLiteModel()
DShapeTestHelper.hpp26 const std::vector<int32_t>& outputTensorShape, in CreateShapeTfLiteModel()
DResizeTestHelper.hpp24 std::vector<char> CreateResizeTfLiteModel(tflite::BuiltinOperator operatorCode, in CreateResizeTfLiteModel()
DQuantizationTestHelper.hpp24 std::vector<char> CreateQuantizationTfLiteModel(tflite::BuiltinOperator quantizationOperatorCode, in CreateQuantizationTfLiteModel()
DLogicalTestHelper.hpp24 std::vector<char> CreateLogicalBinaryTfLiteModel(tflite::BuiltinOperator logicalOperatorCode, in CreateLogicalBinaryTfLiteModel()
DComparisonTestHelper.hpp24 std::vector<char> CreateComparisonTfLiteModel(tflite::BuiltinOperator comparisonOperatorCode, in CreateComparisonTfLiteModel()
DUnpackTestHelper.hpp24 std::vector<char> CreateUnpackTfLiteModel(tflite::BuiltinOperator unpackOperatorCode, in CreateUnpackTfLiteModel()
DBatchMatMulTestHelper.hpp23 std::vector<char> CreateBatchMatMulTfLiteModel( in CreateBatchMatMulTfLiteModel()
DPackTestHelper.hpp24 std::vector<char> CreatePackTfLiteModel(tflite::BuiltinOperator packOperatorCode, in CreatePackTfLiteModel()
DArgMinMaxTestHelper.hpp29 const std::vector<int32_t>& outputTensorShape, in CreateArgMinMaxTfLiteModel()
DReduceTestHelper.hpp24 std::vector<char> CreateReduceTfLiteModel(tflite::BuiltinOperator reduceOperatorCode, in CreateReduceTfLiteModel()
DBatchSpaceTestHelper.hpp24 std::vector<char> CreateBatchSpaceTfLiteModel(tflite::BuiltinOperator batchSpaceOperatorCode, in CreateBatchSpaceTfLiteModel()
DRedefineTestHelper.hpp28 const std::vector<int32_t>& outputTensorShape, in CreateRedefineTfLiteModel()
DElementwiseBinaryTestHelper.hpp25 std::vector<char> CreateElementwiseBinaryTfLiteModel(tflite::BuiltinOperator binaryOperatorCode, in CreateElementwiseBinaryTfLiteModel()
DPadTestHelper.hpp31 const std::vector<int32_t>& outputTensorShape, in CreatePadTfLiteModel()
DPooling2dTestHelper.hpp24 std::vector<char> CreatePooling2dTfLiteModel( in CreatePooling2dTfLiteModel()
DNormalizationTestHelper.hpp27 const std::vector<int32_t>& outputTensorShape, in CreateNormalizationTfLiteModel()
/external/armnn/src/armnnUtils/
DParserHelper.cpp93 … armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateReducedOutputTensoInfo() local
129 armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateStridedSliceOutputTensorInfo() local

12