aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite/Tensor.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/tflite/Tensor.py')
-rw-r--r--ethosu/vela/tflite/Tensor.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tflite/Tensor.py b/ethosu/vela/tflite/Tensor.py
index 4284319a..2d7f9b6c 100644
--- a/ethosu/vela/tflite/Tensor.py
+++ b/ethosu/vela/tflite/Tensor.py
@@ -81,7 +81,7 @@ class Tensor(object):
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
if o != 0:
x = self._tab.Indirect(o + self._tab.Pos)
- from tflite.QuantizationParameters import QuantizationParameters
+ from .QuantizationParameters import QuantizationParameters
obj = QuantizationParameters()
obj.Init(self._tab.Bytes, x)
return obj
@@ -99,7 +99,7 @@ class Tensor(object):
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
if o != 0:
x = self._tab.Indirect(o + self._tab.Pos)
- from tflite.SparsityParameters import SparsityParameters
+ from .SparsityParameters import SparsityParameters
obj = SparsityParameters()
obj.Init(self._tab.Bytes, x)
return obj