Skip to content

Commit 5221745

Browse files
Eli Stevenssoumith
authored andcommitted
add test for bias=False for 3d convolution
1 parent 000ca44 commit 5221745

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test/test_nn.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1631,6 +1631,13 @@ def add_test(test):
16311631
input_size=(2, 3, 3, 4, 5),
16321632
cudnn=True,
16331633
),
1634+
dict(
1635+
module_name='Conv3d',
1636+
constructor_args=(3, 4, (2, 3, 4), 1, 0, 1, 1, False),
1637+
input_size=(2, 3, 3, 4, 5),
1638+
cudnn=True,
1639+
desc='no_bias'
1640+
),
16341641
dict(
16351642
module_name='Conv3d',
16361643
constructor_args=(3, 4, 2, 2),

0 commit comments

Comments
 (0)