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 /SUPPORTED_OPS.md | |
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 'SUPPORTED_OPS.md')
-rw-r--r-- | SUPPORTED_OPS.md | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/SUPPORTED_OPS.md b/SUPPORTED_OPS.md index cee25e7b..43db4c5b 100644 --- a/SUPPORTED_OPS.md +++ b/SUPPORTED_OPS.md @@ -1,7 +1,7 @@ # Supported Ops This file was automatically generated by Vela using the `--supported-ops-report` parameter. -Vela version: `3.6.1.dev1+g30e5320.d20221207` +Vela version: `3.6.1.dev17+g859efbe.d20230203` This file complies with [**Gitiles Markdown syntax**](https://github.com/google/gitiles/blob/master/Documentation/markdown.md) @@ -47,7 +47,7 @@ Please check the supported operator list for your chosen runtime for further inf | SHAPE | [Generic](#tflite-generic-constraints) | | SLICE | [Generic](#tflite-generic-constraints) | | SOFTMAX | [Generic](#tflite-generic-constraints), [Specific](#tflite-softmax-constraints) | -| SPLIT | [Generic](#tflite-generic-constraints) | +| SPLIT | [Generic](#tflite-generic-constraints), [Specific](#tflite-split-constraints) | | SPLIT_V | [Generic](#tflite-generic-constraints), [Specific](#tflite-split_v-constraints) | | SQUEEZE | [Generic](#tflite-generic-constraints), [Specific](#tflite-squeeze-constraints) | | STRIDED_SLICE | [Generic](#tflite-generic-constraints), [Specific](#tflite-strided_slice-constraints) | @@ -286,6 +286,13 @@ This is a list of constraints that the SOFTMAX operator must satisfy in order to - IFM and OFM data types must match - Beta value needs to be positive +### TFLite SPLIT Constraints + +This is a list of constraints that the SPLIT operator must satisfy in order to be scheduled on the NPU. + +- Axis value must be in the range [-RANK(IFM) to +RANK(IFM)) +- Axis must be divisible by number of splits + ### TFLite SPLIT_V Constraints This is a list of constraints that the SPLIT_V operator must satisfy in order to be scheduled on the NPU. |