From 363b572b61f7a32e92cde51478d7556ce43db56f Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Wed, 11 Oct 2023 14:25:50 +0100 Subject: Revert "Revert "MLCE-1093 Reshape and concat invalid results"" This reverts commit 008270f8c1359a7d62c2f881326b4d3f0d8b7b56. Signed-off-by: Mike Kelly Change-Id: If8f5151aa349ff3834f03391e813669e5c51ed66 --- src/armnnTfLiteParser/TfLiteParser.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp') diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 30875d5650..dcd385023b 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -4303,7 +4303,7 @@ void TfLiteParserImpl::ParseUnpack(size_t subgraphIndex, size_t operatorIndex) } splitDesc.SetViewOriginCoord(j, unpackAxis, unpackDimSizes[unpackAxis] * j); } - + splitDesc.SetAxis(unpackAxis); auto layerName = fmt::format("Unpack:{}:{}", subgraphIndex, operatorIndex); IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str()); @@ -4442,7 +4442,10 @@ void TfLiteParserImpl::ParseSplit(size_t subgraphIndex, size_t operatorIndex) } splitDesc.SetViewOriginCoord(j, splitDim, splitterDimSizes[splitDim] * j); } - + if (axisTensorInfo.GetNumElements() == 1) + { + splitDesc.SetAxis(axis); + } auto layerName = fmt::format("Split:{}:{}", subgraphIndex, operatorIndex); IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str()); @@ -4621,6 +4624,7 @@ void TfLiteParserImpl::ParseSplitV(size_t subgraphIndex, size_t operatorIndex) splitDesc.SetViewOriginCoord(j, splitDim, accumSplit); accumSplit += splitSize; } + splitDesc.SetAxis(axis); auto layerName = fmt::format("SplitV:{}:{}", subgraphIndex, operatorIndex); IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str()); -- cgit v1.2.1