diff options
| author | Rickard Bolin <rickard.bolin@arm.com> | 2022-04-27 16:27:31 +0000 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2022-04-27 17:03:36 +0000 |
| commit | 95b07c1c0fed6a985607131e59a593786d40b389 (patch) | |
| tree | 7facce0b4c122ff6c29363420f25e7b4b45f73fe /ethosu/vela/tflite/Model.py | |
| parent | d66f80170e3dc46e6962970b71886010559b1b75 (diff) | |
| download | ethos-u-vela-95b07c1c0fed6a985607131e59a593786d40b389.tar.gz | |
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
Diffstat (limited to 'ethosu/vela/tflite/Model.py')
| -rw-r--r-- | ethosu/vela/tflite/Model.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/ethosu/vela/tflite/Model.py b/ethosu/vela/tflite/Model.py index 660695fb..8a12d103 100644 --- a/ethosu/vela/tflite/Model.py +++ b/ethosu/vela/tflite/Model.py @@ -42,7 +42,7 @@ class Model(object): 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 @@ class Model(object): 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 @@ class Model(object): 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 @@ class Model(object): 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 @@ class Model(object): 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 |