diff options
author | Johan Alfven <johan.alfven@arm.com> | 2023-10-28 16:04:46 +0200 |
---|---|---|
committer | Johan Alfven <johan.alfven@arm.com> | 2023-11-09 11:59:58 +0100 |
commit | a8fda88bced0d11441467b6798885101d41ac8e9 (patch) | |
tree | 807de7fa4eee48720255fbed4a605218f8612f6a /ethosu/vela/tflite_model_semantic.py | |
parent | 4bf0cdf58416edc030ae7507ace95224785e4aa8 (diff) | |
download | ethos-u-vela-a8fda88bced0d11441467b6798885101d41ac8e9.tar.gz |
MLBEDSW-8290: MLCE: Add TRANSPOSE support3.10.0.rc1
- Added graph optimiser function to convert TRANSPOSE op
into an AvgPool op with swapped stride for height and width
- Added TRANSPOSE supported op check
- Added unit tests for TRANSPOSE supported op check
- Updated SUPPORTED_OPS.md
- Fixed problem in pass packing when optimizing the pass list.
Old problem, but now seen when moving TRANSPOSE from cpu.
Change-Id: I0a0ef420b0fb8241090c2e2434622881105cde15
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_model_semantic.py')
-rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_model_semantic.py b/ethosu/vela/tflite_model_semantic.py index 258af93e..d9ace1e8 100644 --- a/ethosu/vela/tflite_model_semantic.py +++ b/ethosu/vela/tflite_model_semantic.py @@ -251,6 +251,9 @@ class TFLiteSemantic: Op.ArgMax: [ TFLiteSemantic.constraint_tens_quant_none_check, ], + Op.Transpose: [ + TFLiteSemantic.constraint_tens_quant_none_check, + ], } return generic_constraints_exclude_list |