From 9f99a4fac1951cdbcc63ae2b1e79184aa9c1e10b Mon Sep 17 00:00:00 2001
From: Wuwei Lin <vincentl13x@gmail.com>
Date: Thu, 30 Aug 2018 00:47:44 -0500
Subject: [PATCH] Fix incorrect doc in conv2d_nhwc_python (#1677)
---
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 461d7a6a0..a872bddab 100644
--- a/topi/python/topi/testing/conv2d_nhwc_python.py
+++ b/topi/python/topi/testing/conv2d_nhwc_python.py
@@ -13,7 +13,7 @@ def conv2d_nhwc_python(a_np, w_np, stride, padding):
4-D with shape [batch, in_height, in_width, in_channel]
w_np : numpy.ndarray
- 4-D with shape [num_filter, filter_height, filter_width, in_channel]
+ 4-D with shape [filter_height, filter_width, in_channel, num_filter]
stride : int or a list/tuple of two ints
Stride size, or [stride_height, stride_width]
--
GitLab