Fix incorrect doc in conv2d_nhwc_python (#1677)

apache · Aug 30, 2018 · 9f99a4f · 9f99a4f
1 parent 7c4dd0b
commit 9f99a4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]