diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-02-14 17:16:12 +0000 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2023-02-20 14:47:07 +0000 |
commit | b2293702c16d107ac1ad80cfac9bd84d804f55d4 (patch) | |
tree | 0da37b0f3759b4f20e6c3b0235e1878b6ba4736f | |
parent | ca361233108b5f0c7d21faf2b97576a7a768866c (diff) | |
download | armnn-b2293702c16d107ac1ad80cfac9bd84d804f55d4.tar.gz |
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 <mike.kelly@arm.com>
Change-Id: I8f03a0400323c865e7e8f924d257d6770f809502
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
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); |