diff options
Diffstat (limited to 'src/core/CL/cl_kernels/direct_convolution_1x1_3x3_5x5_quantized.cl')
-rw-r--r-- | src/core/CL/cl_kernels/direct_convolution_1x1_3x3_5x5_quantized.cl | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/core/CL/cl_kernels/direct_convolution_1x1_3x3_5x5_quantized.cl b/src/core/CL/cl_kernels/direct_convolution_1x1_3x3_5x5_quantized.cl index b58dc7af72..ae87420774 100644 --- a/src/core/CL/cl_kernels/direct_convolution_1x1_3x3_5x5_quantized.cl +++ b/src/core/CL/cl_kernels/direct_convolution_1x1_3x3_5x5_quantized.cl @@ -296,7 +296,12 @@ __kernel void output_stage_quantized( #if defined(HAS_BIAS) // Load and add bias +#if defined(NCHW) int bias_value = *((__global int *)(vector_offset(&bias, get_global_id(2)))); +#else // defined(NCHW) + int16 bias_value = vload16(0, ((__global int *)(vector_offset(&bias, get_global_id(0) * 16)))); +#endif // defined(NCHW) + vals += (int16)(bias_value); #endif //defined(HAS_BIAS) |