commit 9b8fa1222bb24dd36135294b6613c05a25b977c9
author Louis Verhaard <louis.verhaard@arm.com> Fri May 15 13:41:13 2020 +0200
committer Tim Hall <tim.hall@arm.com> Thu Jun 18 17:53:52 2020 +0100
tree 81bc58b60571ba77604630ea7641b4973c922f3d
parent d0eee26bc17ecd237c1b1e86cda78f5f310af391

MLBEDSW-2303: Bug fix Split

Usage of the Split operator caused assert failure.

Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Change-Id: Ibe09b9021f768b86731bdc361f9a0875c4379e4b

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index e28adef..338f962 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -214,7 +214,7 @@
             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]