diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 5 |
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; } |