diff options
| author | Johan Alfven <johan.alfven@arm.com> | 2023-01-31 10:26:26 +0100 |
|---|---|---|
| committer | Johan Alfven <johan.alfven@arm.com> | 2023-02-06 10:09:58 +0100 |
| commit | 12e481147de461e3ea63a8b1dcbc1b66b0fe8e6f (patch) | |
| tree | bbbe6eadb20249fb5974e8753d324b66814d8184 /ethosu/vela/tflite_graph_optimiser.py | |
| parent | 4b7179936d659a6d4abd6b7659c2cc05c5a845fb (diff) | |
| download | ethos-u-vela-12e481147de461e3ea63a8b1dcbc1b66b0fe8e6f.tar.gz | |
MLBEDSW-7284: MLCE: Fix assert for faulty Split op
- An assert in Vela is triggered when the number of splits does
not evenly divide the input.shape[axis] value and the split offsets
are calculated wrongly.
- The fix is to add the same constraints as in the reference kernel
and only run the Split op on the NPU when the criterias are fulfilled.
- Modified test to reflect the new constraints
- Updated SUPPORTED_OPS.md
Change-Id: I4103ff4a3fdf9a813f5fcb7f51081b859e611100
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_graph_optimiser.py')
0 files changed, 0 insertions, 0 deletions