commit 42abec185a95ac8d5f2b32d541f587b3cd306e75
author Tim Hall <tim.hall@arm.com> Thu Feb 04 21:31:57 2021 +0000
committer Tim Hall <tim.hall@arm.com> Fri Feb 05 11:54:48 2021 +0000
tree 7c17eb47cb89a37b9f9d82add61385a5a82c40fe
parent 73e843f76dd71e4ab5e07a7616c2c4806ca6ac25

MLBEDSW-3771: Updated to TF 2.4 flatbuffer schema

Updated tflite loader and mappings from tensorflow 2.3 to tensorflow 2.4

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I55884000ee139baf639bb0377008e0534f72fe94

ethosu/vela/tflite_reader.py

diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py
index 45397c2..ae99c33 100644
--- a/ethosu/vela/tflite_reader.py
+++ b/ethosu/vela/tflite_reader.py
@@ -266,6 +266,8 @@
 
     def parse_operator_code(self, code):
         c = code.BuiltinCode()
+        if c == 0:
+            c = code.DeprecatedBuiltinCode()
         if c not in builtin_operator_map:
             raise InputFileError(
                 self.name, f"The input file contains operator code '{c}' which is currently not supported"