diff options
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),)), |