From 23eccd8c701cbd06c2bc53a12eac90598336ac54 Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Tue, 8 Aug 2023 12:29:14 +0100
Subject: 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 | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ConversionUtils_1_2.hpp b/ConversionUtils_1_2.hpp
index 670463f4..7a629e5a 100644
--- a/ConversionUtils_1_2.hpp
+++ b/ConversionUtils_1_2.hpp
@@ -1122,6 +1122,7 @@ bool ConvertGroupedConv2d(const HalOperation& operation, const HalModel& model,
     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);
-- 
cgit v1.2.1