commit 363b572b61f7a32e92cde51478d7556ce43db56f
author Mike Kelly <mike.kelly@arm.com> Wed Oct 11 14:25:50 2023 +0100
committer mike.kelly <mike.kelly@arm.com> Thu Oct 12 11:57:55 2023 +0000
tree c4f888c607acab8d8e683b77bb20f7c2e507c558
parent c4a932a5f2f8b5f8b684ed2034768136493fc64b

Revert "Revert "MLCE-1093 Reshape and concat invalid results""

This reverts commit 008270f8c1359a7d62c2f881326b4d3f0d8b7b56.

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: If8f5151aa349ff3834f03391e813669e5c51ed66

delegate/opaque/src/Unpack.hpp

diff --git a/delegate/opaque/src/Unpack.hpp b/delegate/opaque/src/Unpack.hpp
index 0956d16..525529f 100644
--- a/delegate/opaque/src/Unpack.hpp
+++ b/delegate/opaque/src/Unpack.hpp
@@ -96,6 +96,8 @@
     unpackDimSizes[unpackAxis] /= unpackNum;
 
     armnn::SplitterDescriptor splitDesc(unpackNum, static_cast<unsigned int>(unpackDimSizes.size()));
+    splitDesc.SetAxis(unpackAxis);
+
     for (unsigned int j = 0; j < unpackNum; ++j)
     {
         // Set the size of the views.