diff options
Diffstat (limited to 'src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl')
-rw-r--r-- | src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl b/src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl index 7c6414312f..4e5481d1db 100644 --- a/src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl +++ b/src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl @@ -177,8 +177,15 @@ __kernel void pooling_3d_layer_MxN_ndhwc( res0 = SQRT_OP(res0); #endif // defined(POOL_L2) - // Store result -#if defined(FP_MIXED_PRECISION) + VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE) + out_q0 = CONVERT(res0, VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE)); + + + + // Store result +#if defined(QUANTIZED) + STORE_VECTOR_SELECT(out_q, DATA_TYPE, out_base_ptr, VEC_SIZE, VEC_SIZE_LEFTOVER, (VEC_SIZE_LEFTOVER != 0) && get_global_id(0) == 0); +#elif defined(FP_MIXED_PRECISION) VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE) res_converted0 = CONVERT(res0, VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE)); STORE_VECTOR_SELECT(res_converted, DATA_TYPE, out_base_ptr, VEC_SIZE, VEC_SIZE_LEFTOVER, (VEC_SIZE_LEFTOVER != 0) && get_global_id(0) == 0); |