diff --git a/include/aidge/backend/cpu/operator/ConvImpl_kernels.hpp b/include/aidge/backend/cpu/operator/ConvImpl_kernels.hpp index e1e76a33120bb9536842a9f0db4cc789f8fe38a1..d2b942f6b6f72235f5d079c0fbb402b1b4ed1373 100644 --- a/include/aidge/backend/cpu/operator/ConvImpl_kernels.hpp +++ b/include/aidge/backend/cpu/operator/ConvImpl_kernels.hpp @@ -631,7 +631,7 @@ void ConvImpl2D_cpu_forward_kernel(const array<DimSize_t, 2> &strideDims, iIndex_channel += inputDims[3] * strideDims[0]) { // loop over associated input line - for (std::size_t ky = 0, ix = 0; ky < kernelDims[0]; + for (std::size_t ky = 0, ix = 0; ky < kernelDims[1]; ++ky, ix += inputDims[3] * dilationDims[0]) { // loop over the entire line for (std::size_t oy = 0, iy = 0; oy < oySize;