From 271ddc31745ca2aefa193f3a7308753126ac7c89 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Tue, 1 Sep 2020 09:15:27 +0200 Subject: MLBEDSW-2814 Add support for inferred size in SplitV For SplitV sizesplit can contain one -1 indicating that dimension is to be inferred. Support added to handle this. Signed-off-by: Patrik Gustavsson Change-Id: Ib9fc8dd2ee1749e81a978d85f2d4a016698bb441 --- ethosu/vela/supported_operators.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'ethosu/vela/supported_operators.py') diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 7cff0ee4..e0ee6163 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -378,6 +378,18 @@ class SupportedOperators: # 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 + if op.type == "SplitV": + # check that maximum one size is set to -1, indicating that size should be inferred + sizes = op.inputs[1].values + num_to_be_inferred = 0 + for size in sizes: + if size == -1: + num_to_be_inferred += 1 + + if num_to_be_inferred > 1: + print("Warning:", op.type, "has more than one size to be inferred, which is illegal, placing on CPU") + return False + return True def check_quantization_restrictions_binary_elem_wise(self, op): -- cgit v1.2.1