From 008270f8c1359a7d62c2f881326b4d3f0d8b7b56 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Arm Date: Fri, 11 Aug 2023 08:55:21 +0000 Subject: Revert "MLCE-1093 Reshape and concat invalid results" This reverts commit 4980e21193f0a14fef084a7f4b4197392f3c0845. Reason for revert: Android Build for v82a failed due to schema not re-generating Change-Id: Ic19cf471b487f321c97ff837d36526512fb12fa4 --- src/backends/aclCommon/ArmComputeUtils.hpp | 16 ++++------------ 1 file changed, 4 insertions(+), 12 deletions(-) (limited to 'src/backends/aclCommon') diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index 9552b7620a..fc59b281b5 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -9,7 +9,6 @@ #include #include #include -#include #include #include @@ -248,20 +247,13 @@ inline std::set ComputeSplitAxis(const armnn::SplitterDescriptor& unsigned int numDimensions = desc.GetNumDimensions(); std::set splitAxis; - if (desc.HasAxis()) + for (unsigned int i = 0; i < numSplit; ++i) { - splitAxis.insert(armnnUtils::GetUnsignedAxis(desc.GetNumDimensions(), desc.GetAxis())); - } - else - { - for (unsigned int i = 0; i < numSplit; ++i) + for (unsigned int dimIdx = 0; dimIdx < numDimensions; ++dimIdx) { - for (unsigned int dimIdx = 0; dimIdx < numDimensions; ++dimIdx) + if (desc.GetViewSizes(i)[dimIdx] != input[dimIdx]) { - if (desc.GetViewSizes(i)[dimIdx] != input[dimIdx]) - { - splitAxis.insert(dimIdx); - } + splitAxis.insert(dimIdx); } } } -- cgit v1.2.1