diff options
| -rw-r--r-- | src/core/CL/cl_kernels/softmax_layer_quantized.cl | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/CL/cl_kernels/softmax_layer_quantized.cl b/src/core/CL/cl_kernels/softmax_layer_quantized.cl index 19cd983cad..31f402f627 100644 --- a/src/core/CL/cl_kernels/softmax_layer_quantized.cl +++ b/src/core/CL/cl_kernels/softmax_layer_quantized.cl @@ -256,8 +256,8 @@ __kernel void softmax_layer_norm_quantized( data = asymm_mult(shifted_scale, data); data = asymm_rounding_divide_by_pow2(data, num_bits_over_unit + 31 - 8); - data = select(0, max(min(data, 255), 0), data_diff >= (int16)(DIFF_MIN)); - vstore16(convert_uchar16(data), 0, (__global uchar *)offset(&dst, 0, 0)); + data = select(0, data, data_diff >= (int16)(DIFF_MIN)); + vstore16(convert_uchar16_sat(data), 0, (__global uchar *)offset(&dst, 0, 0)); } #endif /* defined(DIFF_MIN) */ |