From cf72890e5dd89dada3189816a61d174b984086bd Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Thu, 30 Apr 2020 08:57:23 +0200 Subject: MLBEDSW-1998: Add support for new_axis_mask -Added support for new_axis_mask -Added support for more than 1 bit set in new/shrink_axis mask -Added checks for strided slice in supported operator check -Added assert if nothing has been put on NPU Change-Id: I66e2d04784f14d7ad82371f5d649a455d576a818 Signed-off-by: Patrik Gustavsson --- ethosu/vela/supported_operators.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'ethosu/vela/supported_operators.py') diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 1e11d788..1a25887f 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -240,8 +240,14 @@ class SupportedOperators: return True def check_memory_only_restrictions(self, op): - # check stride size if op.type == "StridedSlice": + # check stride size if len(op.inputs) > 3 and any(stride != 1 for stride in op.inputs[3].values): return False + # check ellipsis_mask + if op.attrs["ellipsis_mask"] != 0: + return False + # check if both new_axis_mask and shrink_axis_mask have bit set + if op.attrs["new_axis_mask"] != 0 and op.attrs["shrink_axis_mask"] != 0: + return False return True -- cgit v1.2.1