diff options
| -rw-r--r-- | ConversionUtils_1_2.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
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); |