diff options
Diffstat (limited to 'src/core/CL/cl_kernels')
-rw-r--r-- | src/core/CL/cl_kernels/winograd_output_transform.cl | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/CL/cl_kernels/winograd_output_transform.cl b/src/core/CL/cl_kernels/winograd_output_transform.cl index 6bd90604e5..f257825967 100644 --- a/src/core/CL/cl_kernels/winograd_output_transform.cl +++ b/src/core/CL/cl_kernels/winograd_output_transform.cl @@ -254,7 +254,7 @@ __kernel void winograd_output_transform_2x2_7x7_nhwc( // Add bias TILE(DATA_TYPE, 1, N0, b); - T_LOAD(DATA_TYPE, 1, N0, BUFFER, bias, cout, 0, 0, b); + T_LOAD(DATA_TYPE, 1, N0, BUFFER, bias, cout, 0, 1, 0, b); T_ADD_BROADCAST_X(DATA_TYPE, 2, N0, out, b, out); #endif // defined(HAS_BIAS) @@ -314,7 +314,7 @@ __kernel void winograd_output_transform_2x2_7x7_nhwc( // Add bias TILE(DATA_TYPE, 1, N0, b); - T_LOAD(DATA_TYPE, 1, N0, BUFFER, bias, cout, 0, 0, b); + T_LOAD(DATA_TYPE, 1, N0, BUFFER, bias, cout, 0, 1, 0, b); T_ADD_BROADCAST_X(DATA_TYPE, 4, N0, out, b, out); #endif // defined(HAS_BIAS) |