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/vela.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/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 63cccc5b..efe8edbb 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -42,7 +42,7 @@ from .tensor import MemArea from .tensor import Tensor from .tflite.Model import Model from .tflite_mapping import builtin_operator_map -from .tflite_mapping import builtin_type_name +from .tflite_mapping import builtin_operator_name_map from .tflite_model_semantic import TFLiteSemantic from .tflite_supported_operators import TFLiteSupportedOperators from .tosa_model_semantic import TosaSemantic @@ -230,7 +230,7 @@ def generate_supported_ops(): raise ValueError op_constraint_links = [] - op_list = sorted(((op, builtin_type_name(op)) for op in builtin_operator_map), key=lambda x: x[1]) + op_list = sorted(((op, builtin_operator_name_map[op]) for op in builtin_operator_map), key=lambda x: x[1]) for op, name in op_list: internal_op = builtin_operator_map[op][0] if internal_op in TFLiteSupportedOperators.supported_operators: |