commit | e631681da8b8f9d69377839466a8970c9ce0f8a1 | [log] [tgz] |
---|---|---|
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | Mon Apr 20 17:09:24 2020 +0100 |
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | Mon Apr 20 18:05:28 2020 +0000 |
tree | 7d849032cd36cbf6b87c7b3346aad4ff79a9d14d | |
parent | 30b46a660629ccd5bf715f63b90d6d7655416e2c [diff] [blame] |
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 b74ad65..61d41b4 100644 --- a/src/runtime/NEON/functions/NEConcatenateLayer.cpp +++ b/src/runtime/NEON/functions/NEConcatenateLayer.cpp
@@ -178,7 +178,7 @@ { for(auto &kernel : _concat_kernels) { - NEScheduler::get().schedule(kernel.get(), _axis); + NEScheduler::get().schedule(kernel.get(), Window::DimY); } } } // namespace arm_compute