diff options
| author | Johan Alfven <johan.alfven@arm.com> | 2023-05-07 13:12:37 +0200 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-06-14 12:53:30 +0000 |
| commit | 8e525ca8aad4a52b80c1986c5067b9b74fb3e321 (patch) | |
| tree | 85f0a781d011aaf1c7631d34acea4398502a7a9f /ethosu/vela/tflite_mapping.py | |
| parent | c2c3063d05494f3968a7fead1b3118602fe100b9 (diff) | |
| download | ethos-u-vela-8e525ca8aad4a52b80c1986c5067b9b74fb3e321.tar.gz | |
MLBEDSW-7748: Add RSQRT support
- Added RSQRT int8 support, implemented as LUT.
- Added test to supported operators
- Updated SUPPORTED_OPS.md
Change-Id: I34904772e044be8d22a6dfe426edf85358a205b7
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
| -rw-r--r-- | ethosu/vela/tflite_mapping.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index dda418c9..83c55bb2 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -793,7 +793,7 @@ builtin_operator_map = { BuiltinOperator.LOG: (Op.Log, None, TFLITE_NO_INDICES), BuiltinOperator.SUM: (Op.Sum, OptionsSerializer("ReducerOptions", ("keep_dims",)), TFLITE_NO_INDICES), BuiltinOperator.SQRT: (Op.Sqrt, None, TFLITE_NO_INDICES), - BuiltinOperator.RSQRT: (Op.Rsqrt, None, TFLITE_NO_INDICES), + BuiltinOperator.RSQRT: (Op.Rsqrt, None, TFLITE_IFM_INDICES), BuiltinOperator.SHAPE: ( Op.Shape, OptionsSerializer("ShapeOptions", (("out_type", datatype_deserialize, datatype_serialize),)), |