diff options
author | Dwight Lidman <dwight.lidman@arm.com> | 2020-08-19 15:53:46 +0200 |
---|---|---|
committer | Dwight Lidman <dwight.lidman@arm.com> | 2020-08-20 09:27:31 +0200 |
commit | a3aad93eec911f6bce69d7fb645d72ed82e59e16 (patch) | |
tree | d3614cc65feb6535f795faebd398f1b6ee89ae7b | |
parent | b16dccec7e03a24f99e63667340696c7fc880d43 (diff) | |
download | ethos-u-vela-a3aad93eec911f6bce69d7fb645d72ed82e59e16.tar.gz |
MLBEDSW-2824: Add mapping for ROUND operator
This commit adds an entry in the tflite_mapping.py
for the ROUND operator, which was previously missing.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: I22d6c60969eea6a785366c6741893718ba3cb8ae
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index e5c3e4f6..adf3462b 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -648,6 +648,7 @@ builtin_operator_map = { BuiltinOperator.MATRIX_DIAG: ("MatrixDiag", OptionsSerializer("MatrixDiagOptions")), BuiltinOperator.QUANTIZE: ("Quantize", OptionsSerializer("QuantizeOptions")), BuiltinOperator.MATRIX_SET_DIAG: ("MatrixSetDiag", OptionsSerializer("MatrixSetDiagOptions")), + BuiltinOperator.ROUND: ("Round", None), BuiltinOperator.HARD_SWISH: ("HardSwish", OptionsSerializer("HardSwishOptions")), BuiltinOperator.IF: ("If", OptionsSerializer("IfOptions", ("then_subgraph_index", "else_subgraph_index"))), BuiltinOperator.WHILE: ("While", OptionsSerializer("WhileOptions", ("cond_subgraph_index", "body_subgraph_index"))), |