From 42abec185a95ac8d5f2b32d541f587b3cd306e75 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 4 Feb 2021 21:31:57 +0000 Subject: 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 Change-Id: I55884000ee139baf639bb0377008e0534f72fe94 --- ethosu/vela/tflite_reader.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'ethosu/vela/tflite_reader.py') diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index 45397c2b..ae99c339 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -266,6 +266,8 @@ class TFLiteGraph: 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" -- cgit v1.2.1