commit 95b07c1c0fed6a985607131e59a593786d40b389
author Rickard Bolin <rickard.bolin@arm.com> Wed Apr 27 16:27:31 2022 +0000
committer Rickard Bolin <rickard.bolin@arm.com> Wed Apr 27 17:03:36 2022 +0000
tree 7facce0b4c122ff6c29363420f25e7b4b45f73fe
parent d66f80170e3dc46e6962970b71886010559b1b75

MLBEDSW-6425: Update to TensorFlow 2.8 (bugfix)

Generate flatbuffer files with relative imports.

Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Idd59bb2ebb829bc42677920577c1f8a04e23ca68

ethosu/vela/tflite/Model.py

diff --git a/ethosu/vela/tflite/Model.py b/ethosu/vela/tflite/Model.py
index 660695f..8a12d10 100644
--- a/ethosu/vela/tflite/Model.py
+++ b/ethosu/vela/tflite/Model.py
@@ -42,7 +42,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.OperatorCode import OperatorCode
+            from .OperatorCode import OperatorCode
             obj = OperatorCode()
             obj.Init(self._tab.Bytes, x)
             return obj
@@ -67,7 +67,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.SubGraph import SubGraph
+            from .SubGraph import SubGraph
             obj = SubGraph()
             obj.Init(self._tab.Bytes, x)
             return obj
@@ -99,7 +99,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.Buffer import Buffer
+            from .Buffer import Buffer
             obj = Buffer()
             obj.Init(self._tab.Bytes, x)
             return obj
@@ -151,7 +151,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.Metadata import Metadata
+            from .Metadata import Metadata
             obj = Metadata()
             obj.Init(self._tab.Bytes, x)
             return obj
@@ -176,7 +176,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.SignatureDef import SignatureDef
+            from .SignatureDef import SignatureDef
             obj = SignatureDef()
             obj.Init(self._tab.Bytes, x)
             return obj