diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-04-20 17:09:24 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-04-20 18:05:28 +0000 |
| commit | e631681da8b8f9d69377839466a8970c9ce0f8a1 (patch) | |
| tree | 7d849032cd36cbf6b87c7b3346aad4ff79a9d14d | |
| parent | 30b46a660629ccd5bf715f63b90d6d7655416e2c (diff) | |
| download | ComputeLibrary-e631681da8b8f9d69377839466a8970c9ce0f8a1.tar.gz | |
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>
| -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 |