commit 4bb989b0e0f82ef2bd25c6848cebf51d253e22ae
author Fredrik Svedberg <fredrik.svedberg@arm.com> Mon Nov 16 15:32:25 2020 +0100
committer patrik.gustavsson <patrik.gustavsson@arm.com> Thu Nov 19 09:28:25 2020 +0000
tree 771e09a7a40fa5bf766e3b691701dcf65edcb684
parent a648aa9928ba6bd0cca5a6de33c2a3cbbee9a0ed

[MLBEDSW-3348] Fix Reshape operator fails with TypeError during deserialization

Fixed Reshape operator fails with TypeError during deserialization
in some cases.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: Ib34142f64295de4524e52a7a28eb36e503047bc0

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index ea9e8a3..20521e4 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -367,8 +367,12 @@
                 if is_vector:
                     fun += "AsNumpy"
 
-                a = deserialize(getattr(tfattrs, fun)())
-                attrs[underscore_mem] = a
+                attr = getattr(tfattrs, fun)()
+                try:
+                    attrs[underscore_mem] = deserialize(attr)
+                except TypeError:
+                    print("Warning: {0} could not read attribute '{1}'.".format(self.name, underscore_mem))
+
         return attrs
 
     def serialize(self, builder, attrs):