From 59b9ab9121d17793b5a240f7c51028b6b37a7a6e Mon Sep 17 00:00:00 2001 From: Raul Farkas Date: Thu, 9 Feb 2023 10:03:27 +0000 Subject: MLBEDSW-7331: Reinstate max stride height constraint of 3 for Conv2D Reinstate constraint for stride height to (1,3) instead of (1,4) for Conv2D and update unit tests. Change-Id: I17389ee040eeff0cea08279cab1c038e951569ea Signed-off-by: Raul Farkas --- ethosu/vela/test/test_tflite_supported_operators.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) (limited to 'ethosu/vela/test') diff --git a/ethosu/vela/test/test_tflite_supported_operators.py b/ethosu/vela/test/test_tflite_supported_operators.py index efe0d000..2713adf9 100644 --- a/ethosu/vela/test/test_tflite_supported_operators.py +++ b/ethosu/vela/test/test_tflite_supported_operators.py @@ -107,7 +107,18 @@ def test_constraint_conv_pass(): @pytest.mark.parametrize( "stride_w, stride_h, supported", - [[0, 20, False], [4, 4, True], [4, 5, False], [5, 4, False], [3, 3, True], [1, 1, True], [2, 4, True]], + [ + [0, 20, False], + [4, 1, True], + [4, 2, True], + [2, 2, True], + [4, 4, False], + [4, 5, False], + [5, 4, False], + [3, 3, True], + [1, 1, True], + [2, 4, False], + ], ) def test_constraint_stride_range(stride_w: int, stride_h: int, supported: bool): # Stride width and height must lie within a certain range -- cgit v1.2.1