diff --git a/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py b/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py index 86511554ea7411a556bf1af0718af5e30a99841d..b5fc85caf274edc9e7afc52df962862fa8a99ba3 100644 --- a/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py +++ b/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py @@ -127,7 +127,7 @@ def prepare_inputs(input_tensor): # input datatype @pytest.mark.parametrize("idt", [DataType.BIPOLAR, DataType.INT2]) # kernel size -@pytest.mark.parametrize("k", [2, 3, 4]) +@pytest.mark.parametrize("k", [2, 3]) # input dimension @pytest.mark.parametrize("ifm_dim", [6, 8]) # input channels