diff options
-rw-r--r-- | src/runtime/NEON/functions/NEConcatenateLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEConcatenateLayer.cpp b/src/runtime/NEON/functions/NEConcatenateLayer.cpp index b74ad6500f..61d41b44f8 100644 --- a/src/runtime/NEON/functions/NEConcatenateLayer.cpp +++ b/src/runtime/NEON/functions/NEConcatenateLayer.cpp @@ -178,7 +178,7 @@ void NEConcatenateLayer::run() { for(auto &kernel : _concat_kernels) { - NEScheduler::get().schedule(kernel.get(), _axis); + NEScheduler::get().schedule(kernel.get(), Window::DimY); } } } // namespace arm_compute |