diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-10-11 14:25:50 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2023-10-12 11:57:55 +0000 |
commit | 363b572b61f7a32e92cde51478d7556ce43db56f (patch) | |
tree | c4f888c607acab8d8e683b77bb20f7c2e507c558 /delegate/opaque/src/Unpack.hpp | |
parent | c4a932a5f2f8b5f8b684ed2034768136493fc64b (diff) | |
download | armnn-363b572b61f7a32e92cde51478d7556ce43db56f.tar.gz |
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
Diffstat (limited to 'delegate/opaque/src/Unpack.hpp')
-rw-r--r-- | delegate/opaque/src/Unpack.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/delegate/opaque/src/Unpack.hpp b/delegate/opaque/src/Unpack.hpp index 0956d1688e..525529ff7b 100644 --- a/delegate/opaque/src/Unpack.hpp +++ b/delegate/opaque/src/Unpack.hpp @@ -96,6 +96,8 @@ TfLiteStatus VisitUnpackOperator(DelegateData& delegateData, 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. |