Fix incorrect doc in conv2d_nhwc_python (#1677)

9f99a4fa · Wuwei Lin · Yizhi Liu · 7c4dd0bd · 9f99a4fa
Commit 9f99a4fa authored Aug 30, 2018 by Wuwei Lin Committed by Yizhi Liu Aug 29, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

topi/python/topi/testing/conv2d_nhwc_python.py
+1 -1

No files found.
--- 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]