diff options
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 8a53039a..bbd44b0b 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -689,3 +689,11 @@ builtin_operator_map = { builtin_operator_inv_map = {v[0]: (k, v[1]) for k, v in builtin_operator_map.items()} builtin_operator_inv_map[Op.CustomNpuOp] = (BuiltinOperator.CUSTOM, CustomOptionsSerializer()) + + +def optype_to_builtintype(op_type): + if op_type in builtin_operator_inv_map: + builtin_type = builtin_operator_inv_map[op_type][0] + return next(k for k, v in vars(BuiltinOperator).items() if v == builtin_type) + else: + return "UNKNOWN" |