diff options
Diffstat (limited to 'src/core/NEON/kernels/NEQuantizationLayerKernel.cpp')
| -rw-r--r-- | src/core/NEON/kernels/NEQuantizationLayerKernel.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/src/core/NEON/kernels/NEQuantizationLayerKernel.cpp b/src/core/NEON/kernels/NEQuantizationLayerKernel.cpp index bff79f0f0c..767af08d0d 100644 --- a/src/core/NEON/kernels/NEQuantizationLayerKernel.cpp +++ b/src/core/NEON/kernels/NEQuantizationLayerKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -77,13 +77,11 @@ void NEQuantizationLayerKernel::run(const Window &window, const ThreadInfo &info ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(INEKernel::window(), window); Window window_input_output(window); - window_input_output.collapse_if_possible(INEKernel::window(), 3); window_input_output.set(3, Window::Dimension(0, 1, 1)); Window window_min_max; window_min_max.use_tensor_dimensions(_min_max->info()->tensor_shape()); window_min_max.set(Window::DimX, Window::Dimension(0, 1, 1)); - window_min_max.collapse_if_possible(INEKernel::window(), 1); Iterator input(_input, window_input_output); Iterator output(_output, window_input_output); |