diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-08-08 12:29:14 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2023-08-09 14:58:02 +0000 |
commit | 23eccd8c701cbd06c2bc53a12eac90598336ac54 (patch) | |
tree | 9394013a8adff3bb2871f879f345aad32911427b /ConversionUtils_1_2.hpp | |
parent | 2c37fb7647839a059188c92c3720df145d618c74 (diff) | |
download | android-nn-driver-23eccd8c701cbd06c2bc53a12eac90598336ac54.tar.gz |
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
Diffstat (limited to 'ConversionUtils_1_2.hpp')
-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); |