diff options
Diffstat (limited to 'src/core')
| -rw-r--r-- | src/core/CL/cl_kernels/depthwise_convolution_quantized.cl | 3 | ||||
| -rw-r--r-- | src/core/CL/kernels/CLROIPoolingLayerKernel.cpp | 4 |
2 files changed, 3 insertions, 4 deletions
diff --git a/src/core/CL/cl_kernels/depthwise_convolution_quantized.cl b/src/core/CL/cl_kernels/depthwise_convolution_quantized.cl index 8a757fc2bd..450342ddfc 100644 --- a/src/core/CL/cl_kernels/depthwise_convolution_quantized.cl +++ b/src/core/CL/cl_kernels/depthwise_convolution_quantized.cl @@ -171,9 +171,8 @@ inline uchar8 convolution3x3( pixels = ASYMM_MULT_BY_QUANT_MULTIPLIER_LESS_THAN_ONE(pixels, output_multiplier, output_shift, 8); pixels = pixels + output_offset; - pixels = clamp(pixels, 0, 255); - return CONVERT(pixels, uchar8); + return CONVERT_SAT(pixels, uchar8); } /** This function computes the horizontal integral of the image. diff --git a/src/core/CL/kernels/CLROIPoolingLayerKernel.cpp b/src/core/CL/kernels/CLROIPoolingLayerKernel.cpp index 4e000c61b1..a07a42408a 100644 --- a/src/core/CL/kernels/CLROIPoolingLayerKernel.cpp +++ b/src/core/CL/kernels/CLROIPoolingLayerKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -108,7 +108,7 @@ void CLROIPoolingLayerKernel::run(const Window &window, cl::CommandQueue &queue) ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(IKernel::window(), window); - Window slice(window); + Window slice = window.first_slice_window_3D(); // Parallelize spatially and across the fourth dimension of the output tensor (also across ROIArray) slice.set(Window::DimZ, window[3]); |