diff options
Diffstat (limited to 'ethosu/vela/test/test_tflite_supported_operators.py')
-rw-r--r-- | ethosu/vela/test/test_tflite_supported_operators.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ethosu/vela/test/test_tflite_supported_operators.py b/ethosu/vela/test/test_tflite_supported_operators.py index e5cc280b..d6b94789 100644 --- a/ethosu/vela/test/test_tflite_supported_operators.py +++ b/ethosu/vela/test/test_tflite_supported_operators.py @@ -529,6 +529,7 @@ def create_strided_slice(): op = create_strided_slice_op([1, 10, 10, 10], [1, 5, 5, 10], [127, 2, 2, 0], [0, 7, -3, 0]) op.attrs["begin_mask"] = 1 op.attrs["end_mask"] = 9 + op.attrs["offset"] = False assert support.is_operator_supported(op) return op @@ -540,6 +541,13 @@ def test_constraint_stridedslice_stride_values(): assert not support.is_operator_supported(op) +def test_constraint_stridedslice_offset_false(): + # Offset attribute must be False + op = create_strided_slice() + op.attrs["offset"] = True + assert not support.is_operator_supported(op) + + def test_constraint_inputs_int32(): # both inputs must be type int32 op = testutil.create_elemwise_op(Op.SHL, "op", [1, 8, 8, 8], [1, 8, 8, 8], [1, 8, 8, 8]) |