COMPMID: Fixed issue in the scheduler of NEConcatenateLayer

Change-Id: Ibee18503ae94da43811e8ad73760abf111c34ff7
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3063
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

diff --git a/src/runtime/NEON/functions/NEConcatenateLayer.cpp b/src/runtime/NEON/functions/NEConcatenateLayer.cpp
index b74ad6500f903f0cb4c6df802a118516081cfda6..61d41b44f87cdb0f8c11c78b189020ea8ece5a82 100644 (file)
--- 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