aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/test
diff options
context:
space:
mode:
authorerik.andersson@arm.com <erik.andersson@arm.com>2021-10-28 14:08:52 +0200
committererik.andersson@arm.com <erik.andersson@arm.com>2021-10-29 13:29:16 +0200
commitba2555e9fe9a93a308cc4e50f6a8a160571cb563 (patch)
treeb0ca8dd14103a4fd7aa9248ddab364986e0ad92e /ethosu/vela/test
parent0af0d383925968626a7c37b084ad806742511e79 (diff)
downloadethos-u-vela-ba2555e9fe9a93a308cc4e50f6a8a160571cb563.tar.gz
MLBEDSW-4925: Fix resize bilinear attribute check
Previously we did not check if half_pixel_centers was set. Since we do not support it, these cases should not run on the NPU. Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com> Change-Id: I9d2675f760424d5cfb67e5d581dd1861ad165b85
Diffstat (limited to 'ethosu/vela/test')
-rw-r--r--ethosu/vela/test/test_tflite_supported_operators.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/ethosu/vela/test/test_tflite_supported_operators.py b/ethosu/vela/test/test_tflite_supported_operators.py
index af5dc174..e3db7913 100644
--- a/ethosu/vela/test/test_tflite_supported_operators.py
+++ b/ethosu/vela/test/test_tflite_supported_operators.py
@@ -327,6 +327,13 @@ def test_constraint_resize():
assert not support.is_operator_supported(op)
+def test_constraint_bilinear_resize_attrs():
+ op = testutil.create_op_with_quant_tensors(Op.ResizeBilinear, [1, 1, 1, 8], [1, 8, 8, 8])
+ assert support.is_operator_supported(op)
+ op.attrs["half_pixel_centers"] = True
+ assert not support.is_operator_supported(op)
+
+
def test_constraint_concat_pass():
# A working concat
op = testutil.create_op_with_quant_tensors(Op.Concat, [1, 1, 1, 4], [1, 1, 1, 8])