From 98d6b3d7e9eef724c6ed64c85c23cd1ad04d7c5a Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Mon, 21 Oct 2019 10:52:54 +0100 Subject: IVGCVSW-3669 Fix TfLiteParser to set Splitter layer outputs' quantization params Signed-off-by: Francis Murtagh Change-Id: I6c69f1f2f0f9c49ddada57b41d5802919ea14b25 --- src/armnnTfLiteParser/TfLiteParser.cpp | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp') diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 24fe7dad0b..e04f9adc9f 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -2276,13 +2276,10 @@ void TfLiteParser::ParseSplit(size_t subgraphIndex, size_t operatorIndex) auto inputTensorIndexes = AsUnsignedVector(GetInputTensorIds(m_Model, subgraphIndex, operatorIndex)); RegisterInputSlots(subgraphIndex, operatorIndex, layer, {inputTensorIndexes[1]}); - TensorShape outShape = TensorShape(static_cast(splitterDimSizes.size()), - splitterDimSizes.data()); - for (unsigned int k = 0; k < layer->GetNumOutputSlots(); ++k) { - layer->GetOutputSlot(k).SetTensorInfo(armnn::TensorInfo(outShape, - inputTensorInfo.GetDataType())); + armnn::TensorInfo tensorInfo = ToTensorInfo(outputs[k]); + layer->GetOutputSlot(k).SetTensorInfo(tensorInfo); } auto outputTensorIndexes = AsUnsignedVector(GetOutputTensorIds(m_Model, subgraphIndex, operatorIndex)); -- cgit v1.2.1