diff options
Diffstat (limited to 'ethosu/vela/tflite/Tensor.py')
-rw-r--r-- | ethosu/vela/tflite/Tensor.py | 88 |
1 files changed, 50 insertions, 38 deletions
diff --git a/ethosu/vela/tflite/Tensor.py b/ethosu/vela/tflite/Tensor.py index 335599e5..4284319a 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 .QuantizationParameters import QuantizationParameters + from tflite.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 .SparsityParameters import SparsityParameters + from tflite.SparsityParameters import SparsityParameters obj = SparsityParameters() obj.Init(self._tab.Bytes, x) return obj @@ -132,39 +132,51 @@ class Tensor(object): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) return o == 0 -def TensorStart(builder): builder.StartObject(8) -def Start(builder): - return TensorStart(builder) -def TensorAddShape(builder, shape): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0) -def AddShape(builder, shape): - return TensorAddShape(builder, shape) -def TensorStartShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def StartShapeVector(builder, numElems): - return TensorStartShapeVector(builder, numElems) -def TensorAddType(builder, type): builder.PrependInt8Slot(1, type, 0) -def AddType(builder, type): - return TensorAddType(builder, type) -def TensorAddBuffer(builder, buffer): builder.PrependUint32Slot(2, buffer, 0) -def AddBuffer(builder, buffer): - return TensorAddBuffer(builder, buffer) -def TensorAddName(builder, name): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0) -def AddName(builder, name): - return TensorAddName(builder, name) -def TensorAddQuantization(builder, quantization): builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(quantization), 0) -def AddQuantization(builder, quantization): - return TensorAddQuantization(builder, quantization) -def TensorAddIsVariable(builder, isVariable): builder.PrependBoolSlot(5, isVariable, 0) -def AddIsVariable(builder, isVariable): - return TensorAddIsVariable(builder, isVariable) -def TensorAddSparsity(builder, sparsity): builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(sparsity), 0) -def AddSparsity(builder, sparsity): - return TensorAddSparsity(builder, sparsity) -def TensorAddShapeSignature(builder, shapeSignature): builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(shapeSignature), 0) -def AddShapeSignature(builder, shapeSignature): - return TensorAddShapeSignature(builder, shapeSignature) -def TensorStartShapeSignatureVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def StartShapeSignatureVector(builder, numElems): - return TensorStartShapeSignatureVector(builder, numElems) -def TensorEnd(builder): return builder.EndObject() -def End(builder): - return TensorEnd(builder)
\ No newline at end of file +def Start(builder): builder.StartObject(8) +def TensorStart(builder): + """This method is deprecated. Please switch to Start.""" + return Start(builder) +def AddShape(builder, shape): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0) +def TensorAddShape(builder, shape): + """This method is deprecated. Please switch to AddShape.""" + return AddShape(builder, shape) +def StartShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def TensorStartShapeVector(builder, numElems): + """This method is deprecated. Please switch to Start.""" + return StartShapeVector(builder, numElems) +def AddType(builder, type): builder.PrependInt8Slot(1, type, 0) +def TensorAddType(builder, type): + """This method is deprecated. Please switch to AddType.""" + return AddType(builder, type) +def AddBuffer(builder, buffer): builder.PrependUint32Slot(2, buffer, 0) +def TensorAddBuffer(builder, buffer): + """This method is deprecated. Please switch to AddBuffer.""" + return AddBuffer(builder, buffer) +def AddName(builder, name): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0) +def TensorAddName(builder, name): + """This method is deprecated. Please switch to AddName.""" + return AddName(builder, name) +def AddQuantization(builder, quantization): builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(quantization), 0) +def TensorAddQuantization(builder, quantization): + """This method is deprecated. Please switch to AddQuantization.""" + return AddQuantization(builder, quantization) +def AddIsVariable(builder, isVariable): builder.PrependBoolSlot(5, isVariable, 0) +def TensorAddIsVariable(builder, isVariable): + """This method is deprecated. Please switch to AddIsVariable.""" + return AddIsVariable(builder, isVariable) +def AddSparsity(builder, sparsity): builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(sparsity), 0) +def TensorAddSparsity(builder, sparsity): + """This method is deprecated. Please switch to AddSparsity.""" + return AddSparsity(builder, sparsity) +def AddShapeSignature(builder, shapeSignature): builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(shapeSignature), 0) +def TensorAddShapeSignature(builder, shapeSignature): + """This method is deprecated. Please switch to AddShapeSignature.""" + return AddShapeSignature(builder, shapeSignature) +def StartShapeSignatureVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def TensorStartShapeSignatureVector(builder, numElems): + """This method is deprecated. Please switch to Start.""" + return StartShapeSignatureVector(builder, numElems) +def End(builder): return builder.EndObject() +def TensorEnd(builder): + """This method is deprecated. Please switch to End.""" + return End(builder)
\ No newline at end of file |