diff options
Diffstat (limited to 'src/core/CL/kernels/CLLocallyConnectedMatrixMultiplyKernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLLocallyConnectedMatrixMultiplyKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLLocallyConnectedMatrixMultiplyKernel.cpp b/src/core/CL/kernels/CLLocallyConnectedMatrixMultiplyKernel.cpp index 794a1bc56e..508fb899f1 100644 --- a/src/core/CL/kernels/CLLocallyConnectedMatrixMultiplyKernel.cpp +++ b/src/core/CL/kernels/CLLocallyConnectedMatrixMultiplyKernel.cpp @@ -101,7 +101,7 @@ void CLLocallyConnectedMatrixMultiplyKernel::run(const Window &window, cl::Comma Window slice = window.first_slice_window_2D(); Window matrix_b_window; - matrix_b_window.use_tensor_dimensions(_input1->info()); + matrix_b_window.use_tensor_dimensions(_input1->info()->tensor_shape()); Window slice_matrix_b = matrix_b_window.first_slice_window_3D(); do |