Skip to content

Commit 9f99a4f

Browse files
vinx13yzhliu
authored andcommitted
Fix incorrect doc in conv2d_nhwc_python (#1677)
1 parent 7c4dd0b commit 9f99a4f

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
@@ -13,7 +13,7 @@ def conv2d_nhwc_python(a_np, w_np, stride, padding):
1313
4-D with shape [batch, in_height, in_width, in_channel]
1414
1515
w_np : numpy.ndarray
16-
4-D with shape [num_filter, filter_height, filter_width, in_channel]
16+
4-D with shape [filter_height, filter_width, in_channel, num_filter]
1717
1818
stride : int or a list/tuple of two ints
1919
Stride size, or [stride_height, stride_width]

0 commit comments

Comments
 (0)