diff options
-rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
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] |