From 38203a860b45616270bdd219ac181d51d18ad060 Mon Sep 17 00:00:00 2001
From: Wuwei Lin <vincentl13x@gmail.com>
Date: Tue, 28 Aug 2018 17:01:57 -0500
Subject: [PATCH] Fix incorrect stride in conv2d_nhwc_python (#1670)
---
topi/python/topi/testing/conv2d_nhwc_python.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/topi/python/topi/testing/conv2d_nhwc_python.py b/topi/python/topi/testing/conv2d_nhwc_python.py
index 880088a6f..461d7a6a0 100644
--- a/topi/python/topi/testing/conv2d_nhwc_python.py
+++ b/topi/python/topi/testing/conv2d_nhwc_python.py
@@ -63,5 +63,5 @@ def conv2d_nhwc_python(a_np, w_np, stride, padding):
apad = at[n, c]
out = scipy.signal.convolve2d(
apad, np.rot90(np.rot90(wt[f, c])), mode='valid')
- bt[n, f] += out[::stride, ::stride]
+ bt[n, f] += out[::stride_h, ::stride_w]
return bt.transpose((0, 2, 3, 1))
--
GitLab