diff options
Diffstat (limited to 'ethosu/vela/tflite_graph_optimiser.py')
| -rw-r--r-- | ethosu/vela/tflite_graph_optimiser.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_graph_optimiser.py b/ethosu/vela/tflite_graph_optimiser.py index daaca8dd..99ac24ee 100644 --- a/ethosu/vela/tflite_graph_optimiser.py +++ b/ethosu/vela/tflite_graph_optimiser.py @@ -45,6 +45,7 @@ from .lstm import Lstm from .lut import convert_to_lut from .lut import create_lut_8bit_op from .lut import create_lut_int16_op +from .lut import create_lut_rsqrt_int8_op from .numeric_util import clamp_sigmoid from .numeric_util import full_shape from .numeric_util import round_away_zero @@ -2048,6 +2049,9 @@ def convert_ops_to_lut(op, arch, nng): # Should already be catched in tflite supported ops assert False, f"Unsupported data type {op.ifm.dtype} for {op.type}" + if op.type == Op.Rsqrt: + return create_lut_rsqrt_int8_op(op) + return op |