diff options
Diffstat (limited to 'ethosu/vela/tflite/SubGraph.py')
| -rw-r--r-- | ethosu/vela/tflite/SubGraph.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tflite/SubGraph.py b/ethosu/vela/tflite/SubGraph.py index d93c5b51..7cd97f26 100644 --- a/ethosu/vela/tflite/SubGraph.py +++ b/ethosu/vela/tflite/SubGraph.py @@ -35,7 +35,7 @@ class SubGraph(object): 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 @@ class SubGraph(object): 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 |