diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-10-21 10:52:54 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-10-22 07:59:35 +0000 |
commit | 98d6b3d7e9eef724c6ed64c85c23cd1ad04d7c5a (patch) | |
tree | 38be49f78f4601d678181f04079cd59910591ed4 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | e606b7cd9d8fb6dfed78ba6f1cf289ddab7182cb (diff) | |
download | armnn-98d6b3d7e9eef724c6ed64c85c23cd1ad04d7c5a.tar.gz |
IVGCVSW-3669 Fix TfLiteParser to set Splitter layer outputs' quantization params
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I6c69f1f2f0f9c49ddada57b41d5802919ea14b25
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 7 |
1 files changed, 2 insertions, 5 deletions
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<unsigned int>(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)); |