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 /ethosu/vela/tflite_mapping.py | |
| 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
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
| -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: ( |