Skip to content

Commit 08f3567

Browse files
committed
add strides argument -- required by some versions of TF
1 parent 3c7a848 commit 08f3567

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1888,7 +1888,7 @@ def test_conv2d_batchnorm_fusion(self):
18881888

18891889
def func_conv2d(x):
18901890
kernel = tf.constant(w, dtype=tf.float32, name='k')
1891-
conv = tf.nn.conv2d(x, kernel, padding='VALID')
1891+
conv = tf.nn.conv2d(x, kernel, strides=[1, 1, 1, 1], padding='VALID')
18921892
return conv
18931893

18941894
def func_fusedbn(x):

0 commit comments

Comments
 (0)