diff options
Diffstat (limited to 'ethosu/vela/tflite_graph_optimiser.py')
| -rw-r--r-- | ethosu/vela/tflite_graph_optimiser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_graph_optimiser.py b/ethosu/vela/tflite_graph_optimiser.py index c7fe6cd9..ef6b90b5 100644 --- a/ethosu/vela/tflite_graph_optimiser.py +++ b/ethosu/vela/tflite_graph_optimiser.py @@ -150,7 +150,7 @@ def rewrite_split_ops(tens, arch, nng): read_shape = None else: # the read shape is relative to each start offset - read_shape = [oe - os for oe, os in zip(offset_end, offset_start)] + read_shape = Shape4D([oe - os for oe, os in zip(offset_end, offset_start)]) # For Split the offset cannot be extracted from the tensor so it has to # be calculated from the index of the output tensor |