Skip to content

Commit 252958f

Browse files
committed
Tests for SepConv io_parallel
1 parent 72e4d5d commit 252958f

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

test/pytest/test_sepconv1d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@
1212
keras_conv1d = [SeparableConv1D]
1313
padds_options = ['same', 'valid']
1414
chans_options = ['channels_last']
15-
io_type_options = ['io_stream']
15+
io_type_options = ['io_parallel', 'io_stream']
1616
strides_options = [(1), (2)]
17-
kernel_options = [(1), (3)]
17+
kernel_options = [(2), (3)]
1818
bias_options = [False]
1919

2020

test/pytest/test_sepconv2d.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,19 +12,19 @@
1212
keras_conv2d = [SeparableConv2D]
1313
padds_options = ['same', 'valid']
1414
chans_options = ['channels_last']
15-
io_type_options = ['io_stream']
15+
io_type_options = ['io_parallel', 'io_stream']
1616
strides_options = [(1, 1), (2, 2)]
1717
kernel_options = [(2, 2), (3, 3)]
1818
bias_options = [False]
1919

2020

21-
@pytest.mark.parametrize("conv2d", keras_conv2d)
22-
@pytest.mark.parametrize("chans", chans_options)
23-
@pytest.mark.parametrize("padds", padds_options)
24-
@pytest.mark.parametrize("strides", strides_options)
25-
@pytest.mark.parametrize("kernels", kernel_options)
26-
@pytest.mark.parametrize("bias", bias_options)
27-
@pytest.mark.parametrize("io_type", io_type_options)
21+
@pytest.mark.parametrize('conv2d', keras_conv2d)
22+
@pytest.mark.parametrize('chans', chans_options)
23+
@pytest.mark.parametrize('padds', padds_options)
24+
@pytest.mark.parametrize('strides', strides_options)
25+
@pytest.mark.parametrize('kernels', kernel_options)
26+
@pytest.mark.parametrize('bias', bias_options)
27+
@pytest.mark.parametrize('io_type', io_type_options)
2828
@pytest.mark.parametrize('backend', ['Vivado', 'Vitis', 'Catapult'])
2929
def test_sepconv2d(conv2d, chans, padds, strides, kernels, bias, io_type, backend):
3030
model = tf.keras.models.Sequential()

0 commit comments

Comments
 (0)