diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2017-07-11 18:13:08 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:15:39 +0100 |
| commit | be35b0ed7cb6fbf9010142aab97854c467b14c5c (patch) | |
| tree | bd654b559317e438a6703a09e243e4401d41d07d /src/core/CL/kernels/CLActivationLayerKernel.cpp | |
| parent | 7ff47a313d62011923c5307cc52510b4ba41a631 (diff) | |
| download | ComputeLibrary-be35b0ed7cb6fbf9010142aab97854c467b14c5c.tar.gz | |
COMPMID-443: Collapse higher dimensions for activation layer
Change-Id: I5943235aff1bb6440e3ab08e818d53aa5d94143a
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/80349
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLActivationLayerKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLActivationLayerKernel.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/core/CL/kernels/CLActivationLayerKernel.cpp b/src/core/CL/kernels/CLActivationLayerKernel.cpp index 5e8204fa1b..fda69b0b94 100644 --- a/src/core/CL/kernels/CLActivationLayerKernel.cpp +++ b/src/core/CL/kernels/CLActivationLayerKernel.cpp @@ -99,7 +99,8 @@ void CLActivationLayerKernel::run(const Window &window, cl::CommandQueue &queue) ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); - Window slice = window.first_slice_window_3D(); + Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); + Window slice = collapsed.first_slice_window_3D(); do { @@ -111,5 +112,5 @@ void CLActivationLayerKernel::run(const Window &window, cl::CommandQueue &queue) } enqueue(queue, *this, slice); } - while(window.slide_window_slice_3D(slice)); + while(collapsed.slide_window_slice_3D(slice)); } |