Home
last modified time | relevance | path

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

/external/armnn/delegate/src/test/
DFullyConnectedTest.cpp16 std::vector<int32_t> outputTensorShape { 1, 1 }; in FullyConnectedFp32Test() local
41 std::vector<int32_t> outputTensorShape { 1, 1 }; in FullyConnectedActicationTest() local
66 std::vector<int32_t> outputTensorShape { 2, 1 }; in FullyConnectedInt8Test() local
DFullyConnectedTestHelper.hpp23 std::vector<char> CreateFullyConnectedTfLiteModel(tflite::TensorType tensorType, in CreateFullyConnectedTfLiteModel()
152 void FullyConnectedTest(std::vector<armnn::BackendId>& backends, in FullyConnectedTest()
DControlTestHelper.hpp26 std::vector<char> CreateConcatTfLiteModel(tflite::BuiltinOperator controlOperatorCode, in CreateConcatTfLiteModel()
115 std::vector<char> CreateMeanTfLiteModel(tflite::BuiltinOperator controlOperatorCode, in CreateMeanTfLiteModel()
DConvolutionTestHelper.hpp23 std::vector<char> CreateConv2dTfLiteModel(tflite::BuiltinOperator convolutionOperatorCode, in CreateConv2dTfLiteModel()
285 std::vector<char> CreateTransposeConvTfLiteModel(tflite::TensorType tensorType, in CreateTransposeConvTfLiteModel()
405 void TransposeConvTest(std::vector<armnn::BackendId>& backends, in TransposeConvTest()
DResizeTestHelper.hpp24 std::vector<char> CreateResizeTfLiteModel(tflite::BuiltinOperator operatorCode, in CreateResizeTfLiteModel()
DTransposeTestHelper.hpp21 std::vector<char> CreateTransposeTfLiteModel(tflite::TensorType tensorType, in CreateTransposeTfLiteModel()
DQuantizationTestHelper.hpp22 std::vector<char> CreateQuantizationTfLiteModel(tflite::BuiltinOperator quantizationOperatorCode, in CreateQuantizationTfLiteModel()
DRedefineTestHelper.hpp28 const std::vector<int32_t>& outputTensorShape, in CreateRedefineTfLiteModel()
DComparisonTestHelper.hpp24 std::vector<char> CreateComparisonTfLiteModel(tflite::BuiltinOperator comparisonOperatorCode, in CreateComparisonTfLiteModel()
DPooling2dTestHelper.hpp24 std::vector<char> CreatePooling2dTfLiteModel( in CreatePooling2dTfLiteModel()
DElementwiseBinaryTestHelper.hpp22 std::vector<char> CreateElementwiseBinaryTfLiteModel(tflite::BuiltinOperator binaryOperatorCode, in CreateElementwiseBinaryTfLiteModel()
/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
DParserPrototxtFixture.hpp107const armnn::TensorShape& outputTensorShape, in SetupSingleInputSingleOutput()
/external/armnn/src/armnn/test/
DInferOutputTests.hpp570 armnn::TensorShape outputTensorShape(2, outputShape.data()); in QLstmInferOutputShapeTest() local
618 armnn::TensorShape outputTensorShape(2, outputShape.data()); in QuantizedLstmInferOutputShapeTest() local
/external/armnn/src/armnn/
DGraph.cpp100 auto outputTensorShape = layer->GetOutputSlots()[i].GetTensorInfo().GetShape(); in Print() local