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_writer.py | 1 + 1 file changed, 1 insertion(+) (limited to 'ethosu/vela/tflite_writer.py') diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index 82a063ff..e190a746 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -193,6 +193,7 @@ class TFLiteSerialiser: self.operator_code_map[op_type] = (idx, tf_code, opt_serializer) OperatorCode.OperatorCodeStart(builder) + OperatorCode.OperatorCodeAddDeprecatedBuiltinCode(builder, tf_code if tf_code < 127 else 127) OperatorCode.OperatorCodeAddBuiltinCode(builder, tf_code) if custom_code_offset is not None: OperatorCode.OperatorCodeAddCustomCode(builder, custom_code_offset) -- cgit v1.2.1