|
12 | 12 | keras_conv2d = [SeparableConv2D]
|
13 | 13 | padds_options = ['same', 'valid']
|
14 | 14 | chans_options = ['channels_last']
|
15 |
| -io_type_options = ['io_stream'] |
| 15 | +io_type_options = ['io_parallel', 'io_stream'] |
16 | 16 | strides_options = [(1, 1), (2, 2)]
|
17 | 17 | kernel_options = [(2, 2), (3, 3)]
|
18 | 18 | bias_options = [False]
|
19 | 19 |
|
20 | 20 |
|
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) |
28 | 28 | @pytest.mark.parametrize('backend', ['Vivado', 'Vitis', 'Catapult'])
|
29 | 29 | def test_sepconv2d(conv2d, chans, padds, strides, kernels, bias, io_type, backend):
|
30 | 30 | model = tf.keras.models.Sequential()
|
|
0 commit comments