diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2022-04-21 07:36:55 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2022-04-27 07:49:36 +0000 |
commit | d66f80170e3dc46e6962970b71886010559b1b75 (patch) | |
tree | a8efaa8cb3f07f16758086e29f9114b1ca40e870 /ethosu/vela/tflite/Tensor.py | |
parent | a2ec5aa747633da72b6310ce7e5552c39f7f54bb (diff) | |
download | ethos-u-vela-d66f80170e3dc46e6962970b71886010559b1b75.tar.gz |
MLBEDSW-6425: Update to TensorFlow 2.8
Update the flatbuffers generated code to comply with TensorFlow 2.8
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Ia65325b88745e49dbafa803a38c0ea0e7d0478ba
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 |