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/tosa_mapping.py | |
| parent | 68df8a1f5469daac53b7a418d92204f7026e4228 (diff) | |
| download | ethos-u-vela-dev/mlbedsw-6271.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/tosa_mapping.py')
| -rw-r--r-- | ethosu/vela/tosa_mapping.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/ethosu/vela/tosa_mapping.py b/ethosu/vela/tosa_mapping.py index 6710787e..f635e4a6 100644 --- a/ethosu/vela/tosa_mapping.py +++ b/ethosu/vela/tosa_mapping.py @@ -323,14 +323,12 @@ tosa_operator_map = { tosa_operator_inv_map = {v[0]: (k, v[1]) for k, v in tosa_operator_map.items()} - -def tosa_type_name(builtin): - return next(k for k, v in vars(TosaOp).items() if v == builtin) +tosa_operator_name_map = {v: k for k, v in vars(TosaOp).items()} # TODO will return UNKNOWN for the once that have not yet been defined in tosa_operator_map -def optype_to_tosa_op_type(op_type): +def optype_to_tosa_op_type(op_type: Op): if op_type in tosa_operator_inv_map: - return tosa_type_name(tosa_operator_inv_map[op_type][0]) + return tosa_operator_name_map[tosa_operator_inv_map[op_type][0]] else: return TosaOp.UNKNOWN |