aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite/SignatureDef.py
diff options
context:
space:
mode:
authorRickard Bolin <rickard.bolin@arm.com>2022-04-27 16:27:31 +0000
committerRickard Bolin <rickard.bolin@arm.com>2022-04-27 17:03:36 +0000
commit95b07c1c0fed6a985607131e59a593786d40b389 (patch)
tree7facce0b4c122ff6c29363420f25e7b4b45f73fe /ethosu/vela/tflite/SignatureDef.py
parentd66f80170e3dc46e6962970b71886010559b1b75 (diff)
downloadethos-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/SignatureDef.py')
-rw-r--r--ethosu/vela/tflite/SignatureDef.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tflite/SignatureDef.py b/ethosu/vela/tflite/SignatureDef.py
index 787b6840..f75494b5 100644
--- a/ethosu/vela/tflite/SignatureDef.py
+++ b/ethosu/vela/tflite/SignatureDef.py
@@ -35,7 +35,7 @@ class SignatureDef(object):
x = self._tab.Vector(o)
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
x = self._tab.Indirect(x)
- from tflite.TensorMap import TensorMap
+ from .TensorMap import TensorMap
obj = TensorMap()
obj.Init(self._tab.Bytes, x)
return obj
@@ -60,7 +60,7 @@ class SignatureDef(object):
x = self._tab.Vector(o)
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
x = self._tab.Indirect(x)
- from tflite.TensorMap import TensorMap
+ from .TensorMap import TensorMap
obj = TensorMap()
obj.Init(self._tab.Bytes, x)
return obj