commit 23eccd8c701cbd06c2bc53a12eac90598336ac54
author Mike Kelly <mike.kelly@arm.com> Tue Aug 08 12:29:14 2023 +0100
committer mike.kelly <mike.kelly@arm.com> Wed Aug 09 14:58:02 2023 +0000
tree 9394013a8adff3bb2871f879f345aad32911427b
parent 2c37fb7647839a059188c92c3720df145d618c74

MLCE-1093 Reshape and concat invalid results

!armnn:10062

 * Added Split Axis to GroupedConv2d splitter.

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

ConversionUtils_1_2.hpp

diff --git a/ConversionUtils_1_2.hpp b/ConversionUtils_1_2.hpp
index 670463f..7a629e5 100644
--- a/ConversionUtils_1_2.hpp
+++ b/ConversionUtils_1_2.hpp
@@ -1122,6 +1122,7 @@
     std::vector<std::reference_wrapper<TensorInfo>> splitterOutputInfos(numGroups, std::ref(splitterOutputInfo));
 
     ViewsDescriptor splitterDesc(numGroups);
+    splitterDesc.SetAxis(armnn::numeric_cast<int32_t>(channelsIndex));
     for (unsigned int group = 0u; group < numGroups; ++group)
     {
         splitterDesc.SetViewOriginCoord(group, channelsIndex, splitterDimSizes[channelsIndex] * group);