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/SubGraph.py

diff --git a/ethosu/vela/tflite/SubGraph.py b/ethosu/vela/tflite/SubGraph.py
index d93c5b5..7cd97f2 100644
--- a/ethosu/vela/tflite/SubGraph.py
+++ b/ethosu/vela/tflite/SubGraph.py
@@ -35,7 +35,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.Tensor import Tensor
+            from .Tensor import Tensor
             obj = Tensor()
             obj.Init(self._tab.Bytes, x)
             return obj
@@ -114,7 +114,7 @@
             x = self._tab.Vector(o)
             x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
             x = self._tab.Indirect(x)
-            from tflite.Operator import Operator
+            from .Operator import Operator
             obj = Operator()
             obj.Init(self._tab.Bytes, x)
             return obj