commit 008270f8c1359a7d62c2f881326b4d3f0d8b7b56
author Nikhil Raj Arm <nikhil.raj@arm.com> Fri Aug 11 08:55:21 2023 +0000
committer Nikhil Raj Arm <nikhil.raj@arm.com> Fri Aug 11 08:55:21 2023 +0000
tree 5496cc6b428e1ecc0740aaf62dba331c77a9a9eb
parent 4980e21193f0a14fef084a7f4b4197392f3c0845

Revert "MLCE-1093 Reshape and concat invalid results"

This reverts commit 4980e21193f0a14fef084a7f4b4197392f3c0845.

Reason for revert: Android Build for v82a failed due to schema not re-generating

Change-Id: Ic19cf471b487f321c97ff837d36526512fb12fa4

src/armnnTfLiteParser/TfLiteParser.cpp

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index f40f7ff..301989e 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -4213,7 +4213,7 @@
         }
         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());
 
@@ -4352,10 +4352,7 @@
         }
         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());
 
@@ -4534,7 +4531,6 @@
         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());