commit cfb42620d7e061efdfc92bf944d0289ebfc02ea4
author Dwight Lidman <dwight.lidman@arm.com> Wed Aug 11 12:28:51 2021 +0200
committer Dwight Lidman <dwight.lidman@arm.com> Fri Aug 13 15:11:06 2021 +0100
tree 71b39ca6de27ec05853a6604d722ac4d6d918af9
parent 53c6245a2b07dbe972d664b65faea61acd6a5980

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

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index e0f2493..33117eb 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -567,9 +567,7 @@
     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 @@
     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: (