diff options
Diffstat (limited to 'src/core/CL/ICLKernel.cpp')
-rw-r--r-- | src/core/CL/ICLKernel.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/CL/ICLKernel.cpp b/src/core/CL/ICLKernel.cpp index 1c6963f3f1..9ba17d0e03 100644 --- a/src/core/CL/ICLKernel.cpp +++ b/src/core/CL/ICLKernel.cpp @@ -105,8 +105,8 @@ void ICLKernel::add_tensor_argument(unsigned &idx, const ICLTensor *tensor, cons for(unsigned int d = 0; d < dimension_size; ++d) { - _kernel.setArg<cl_uint>(idx++, strides[d]); - _kernel.setArg<cl_uint>(idx++, strides[d] * window[d].step()); + _kernel.setArg<cl_uint>(idx++, window.is_broadcasted(d) ? 0 : strides[d]); + _kernel.setArg<cl_uint>(idx++, window.is_broadcasted(d) ? 0 : (strides[d] * window[d].step())); } _kernel.setArg<cl_uint>(idx++, offset_first_element); |