aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/test/test_tflite_supported_operators.py
diff options
context:
space:
mode:
authorRickard Bolin <rickard.bolin@arm.com>2023-09-29 12:48:29 +0000
committerRickard Bolin <rickard.bolin@arm.com>2023-10-11 11:07:22 +0000
commitb37a81bfce0c7ae3092e4b3c9f69e08644b84e25 (patch)
tree88f25785f8b978cc57ad4a7b4c83e8cea84033dd /ethosu/vela/test/test_tflite_supported_operators.py
parent529b787f0321cbae1878ecb0994c60f670bc7ea1 (diff)
downloadethos-u-vela-b37a81bfce0c7ae3092e4b3c9f69e08644b84e25.tar.gz
MLBEDSW-8111: Update to TensorFlow 2.14
- Update to TensorFlow 2.14 and minimum required Python version to 3.9. - Update version pins on NumPy and FlatBuffers. - Add constraint to Offset attribute of StridedSlice operator Change-Id: I8c7122def963202e5f47e92b62be607935ed05cf Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Diffstat (limited to 'ethosu/vela/test/test_tflite_supported_operators.py')
-rw-r--r--ethosu/vela/test/test_tflite_supported_operators.py8
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])