Skip to content

Commit cd73fc4

Browse files
vinx13yzhliu
authored andcommitted
Fix incorrect stride in conv2d_nhwc_python (apache#1670)
1 parent bfed21f commit cd73fc4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

topi/python/topi/testing/conv2d_nhwc_python.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,5 +63,5 @@ def conv2d_nhwc_python(a_np, w_np, stride, padding):
6363
apad = at[n, c]
6464
out = scipy.signal.convolve2d(
6565
apad, np.rot90(np.rot90(wt[f, c])), mode='valid')
66-
bt[n, f] += out[::stride, ::stride]
66+
bt[n, f] += out[::stride_h, ::stride_w]
6767
return bt.transpose((0, 2, 3, 1))

0 commit comments

Comments
 (0)