From afb56ae11294bdf50e190cfafd76e507d5e2787d Mon Sep 17 00:00:00 2001 From: Johan Alfven Date: Fri, 27 Oct 2023 13:08:21 +0200 Subject: MLBEDSW-8201: [MLCE] Extended stride support for CONV_2D - Added support for stride_h > 3 when ofm height is 1 - Added support for stride_w > 3 when ofm width is 1 - Updated constraints - Updated tests - Updated SUPPORTED_OPS.md Change-Id: I8f89909b05a0f052df5f03702966cee50da61cfc Signed-off-by: Johan Alfven --- ethosu/vela/test/test_tflite_supported_operators.py | 5 +++++ 1 file changed, 5 insertions(+) (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 d6b94789..a433fb8d 100644 --- a/ethosu/vela/test/test_tflite_supported_operators.py +++ b/ethosu/vela/test/test_tflite_supported_operators.py @@ -128,6 +128,11 @@ def test_constraint_stride_range(ifm_shape: List[int], stride_w: int, stride_h: op = testutil.create_op_with_quant_tensors(Op.Conv2DBias, ifm_shape, [1, 8, 8, 8], [1, 1, 1, 1]) op.attrs = {"stride_w": stride_w, "stride_h": stride_h} assert support.is_operator_supported(op) == supported + if not supported and stride_w > 0 and stride_h > 0: + # Test not supported but with ofm width and height = 1 -> supported + op = testutil.create_op_with_quant_tensors(Op.Conv2DBias, ifm_shape, [1, 1, 1, 8], [1, 1, 1, 1]) + op.attrs = {"stride_w": stride_w, "stride_h": stride_h} + assert support.is_operator_supported(op) def test_constraint_dilated_height_range(): -- cgit v1.2.1