From 4ede26c2c6a7b3bfa3020a7643477da3b3f4531a Mon Sep 17 00:00:00 2001 From: Przemyslaw Tredak Date: Mon, 10 Jul 2017 13:49:08 -0700 Subject: [PATCH] Fix im2col.h (#6986) --- src/operator/nn/im2col.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/operator/nn/im2col.h b/src/operator/nn/im2col.h index 435d502da77e..ce4d9e31db9c 100644 --- a/src/operator/nn/im2col.h +++ b/src/operator/nn/im2col.h @@ -241,7 +241,7 @@ inline void im2col(mshadow::Stream* s, if (2 == kernel_shape.ndim()) { im2col_cpu(data_im, im_shape[1], im_shape[2], im_shape[3], kernel_shape[0], kernel_shape[1], pad[0], pad[1], - stride[0], stride[1], dilation[1], dilation[1], data_col); + stride[0], stride[1], dilation[0], dilation[1], data_col); } else { im2col_nd_core_cpu(data_im, true, im_shape, col_shape, kernel_shape, pad, stride, dilation, data_col);