From cfb42620d7e061efdfc92bf944d0289ebfc02ea4 Mon Sep 17 00:00:00 2001 From: Dwight Lidman Date: Wed, 11 Aug 2021 12:28:51 +0200 Subject: 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 Change-Id: Ia0b577ca44d76486fc3e0ea9780e0dc1d2baf65e --- ethosu/vela/tflite_mapping.py | 9 ++------- 1 file 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: ( -- cgit v1.2.1