From 9b8fa1222bb24dd36135294b6613c05a25b977c9 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Fri, 15 May 2020 13:41:13 +0200 Subject: MLBEDSW-2303: Bug fix Split Usage of the Split operator caused assert failure. Signed-off-by: Louis Verhaard Change-Id: Ibe09b9021f768b86731bdc361f9a0875c4379e4b --- ethosu/vela/operation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela') diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index e28adef6..338f962e 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -214,7 +214,7 @@ input and output tensors, as well as an attribute dictionary.""" outputs = self.outputs assert num_splits == len(outputs) - if self.type == "SplitV": + elif self.type == "SplitV": num_splits = self.attrs.get("num_splits") input_tens = self.inputs[0] size_tens = self.inputs[1] -- cgit v1.2.1