diff options
author | Tim Hall <tim.hall@arm.com> | 2022-03-03 17:43:16 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2022-04-04 14:25:01 +0100 |
commit | a3fe665803c0f72000f9dda249446d5a0d03240f (patch) | |
tree | 1de76663d1a1d5a39cf795eecbc99d5479382735 /ethosu/vela/tflite_mapping.py | |
parent | 68df8a1f5469daac53b7a418d92204f7026e4228 (diff) | |
download | ethos-u-vela-a3fe665803c0f72000f9dda249446d5a0d03240f.tar.gz |
vela: Minor refactordev/mlbedsw-6271
- Changed comments to docstring on QuantizationParams
- Simplified op type to op name conversion
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I2fdf5922cc17944c9bd37917a85fdfe50a1e651d
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 8 |
1 files changed, 3 insertions, 5 deletions
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 |