1 files changed, 2 insertions, 3 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp
index 5fd1202659..3a398e03e6 100644
--- a/src/backends/neon/NeonBackend.cpp
+++ b/src/backends/neon/NeonBackend.cpp
@@ -516,9 +516,8 @@ OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph
             {
                 continue;
             }
-            // Cannot remove a Reshape if it's connected to a SplitterLayer through a Tensor that has more than
-            // 4 dimensions
-            if (ConnectedToLayerType(baseLayer, LayerType::Splitter, 4))
+            // Cannot remove a Reshape if it's connected to a SplitterLayer
+            if (ConnectedToLayerType(baseLayer, LayerType::Splitter))
             {
                 continue;
             }