diff options
author | Dwight Lidman <dwight.lidman@arm.com> | 2021-08-11 12:28:51 +0200 |
---|---|---|
committer | Dwight Lidman <dwight.lidman@arm.com> | 2021-08-13 15:11:06 +0100 |
commit | cfb42620d7e061efdfc92bf944d0289ebfc02ea4 (patch) | |
tree | 71b39ca6de27ec05853a6604d722ac4d6d918af9 | |
parent | 53c6245a2b07dbe972d664b65faea61acd6a5980 (diff) | |
download | ethos-u-vela-cfb42620d7e061efdfc92bf944d0289ebfc02ea4.tar.gz |
MLBEDSW-4993: TFLite mapping error for Wav2Letter
Previous to this commit some networks were failing due
one or more options in the TFLite mapping being
incorrect after the update to match TF 2.5.
This commit reverts those changes.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Ia0b577ca44d76486fc3e0ea9780e0dc1d2baf65e
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index e0f24932..33117eb0 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -567,9 +567,7 @@ builtin_operator_map = { BuiltinOperator.RELU6: (Op.Relu6, None, TFLITE_IFM_INDICES), BuiltinOperator.RESHAPE: ( Op.Reshape, - OptionsSerializer( - "ReshapeOptions", (("new_shape", is_int_vec), "new_shape_as_numpy", "new_shape_is_none", "new_shape_length") - ), + OptionsSerializer("ReshapeOptions", (("new_shape", is_int_vec),)), TFLITE_IFM_INDICES, ), BuiltinOperator.RESIZE_BILINEAR: ( @@ -654,10 +652,7 @@ builtin_operator_map = { BuiltinOperator.DIV: (Op.Div, OptionsSerializer("DivOptions", (fused_act,)), TFLITE_NO_INDICES), BuiltinOperator.SQUEEZE: ( Op.Squeeze, - OptionsSerializer( - "SqueezeOptions", - (("squeeze_dims", is_int_vec), "squeeze_dims_as_numpy", "squeeze_dims_is_none", "squeeze_dims_length"), - ), + OptionsSerializer("SqueezeOptions", (("squeeze_dims", is_int_vec),),), TFLITE_IFM_INDICES, ), BuiltinOperator.UNIDIRECTIONAL_SEQUENCE_LSTM: ( |