From a3fe665803c0f72000f9dda249446d5a0d03240f Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 3 Mar 2022 17:43:16 +0000 Subject: vela: Minor refactor - Changed comments to docstring on QuantizationParams - Simplified op type to op name conversion Signed-off-by: Tim Hall Change-Id: I2fdf5922cc17944c9bd37917a85fdfe50a1e651d --- ethosu/vela/tflite_mapping.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'ethosu/vela/tflite_mapping.py') diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index a5f7fa2b..e12267f0 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -950,13 +950,11 @@ builtin_operator_inv_map[Op.CustomNpuOp] = (BuiltinOperator.CUSTOM, CustomOption BUILTIN_OPERATOR_UNKNOWN = "UNKNOWN" +builtin_operator_name_map = {v: k for k, v in vars(BuiltinOperator).items()} -def builtin_type_name(builtin): - return next(k for k, v in vars(BuiltinOperator).items() if v == builtin) - -def optype_to_builtintype(op_type): +def optype_to_builtintype(op_type: Op): if op_type in builtin_operator_inv_map: - return builtin_type_name(builtin_operator_inv_map[op_type][0]) + return builtin_operator_name_map[builtin_operator_inv_map[op_type][0]] else: return BUILTIN_OPERATOR_UNKNOWN -- cgit v1.2.1