From b2293702c16d107ac1ad80cfac9bd84d804f55d4 Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Tue, 14 Feb 2023 17:16:12 +0000 Subject: IVGCVSW-7427 Execute network unable to correctly determine the shape of the network * Fixed bug where calculated output shape wasn't being recorded in ParseSqueeze. Signed-off-by: Mike Kelly Change-Id: I8f03a0400323c865e7e8f924d257d6770f809502 --- src/armnnTfLiteParser/TfLiteParser.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src') diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 66dc13ef49..9e8af66b49 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -2255,6 +2255,9 @@ void TfLiteParserImpl::ParseSqueeze(size_t subgraphIndex, size_t operatorIndex) ReshapeDescriptor reshapeDesc; reshapeDesc.m_TargetShape = outputTensorInfo.GetShape(); + auto outputTensorIds = GetOutputTensorIds(m_Model, subgraphIndex, operatorIndex); + m_TensorInfos[outputTensorIds[0]] = outputTensorInfo; + IConnectableLayer* layer = m_Network->AddReshapeLayer(reshapeDesc, layerName.c_str()); ARMNN_ASSERT(layer != nullptr); layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); -- cgit v1.2.1