diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2021-12-20 08:35:23 +0000 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2022-01-21 12:45:02 +0000 |
commit | 2de898a85d800f0e812cc13eff6363e27a377cf5 (patch) | |
tree | 4c4b81cf5c4c521d33af59234deffce00611ba78 /ethosu/vela/tflite/QuantizationParameters.py | |
parent | 3c25ff658cf847ccfcb2d4d5796ffbe13a511894 (diff) | |
download | ethos-u-vela-2de898a85d800f0e812cc13eff6363e27a377cf5.tar.gz |
MLBEDSW-4870: Update to TensorFlow 2.7
Update the flatbuffers generated code to comply with TensorFlow 2.7
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Iff29b05a6e145245861329b4ff9fc9fbd968da53
Diffstat (limited to 'ethosu/vela/tflite/QuantizationParameters.py')
-rw-r--r-- | ethosu/vela/tflite/QuantizationParameters.py | 32 |
1 files changed, 31 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/QuantizationParameters.py b/ethosu/vela/tflite/QuantizationParameters.py index 50acb835..0a27af0e 100644 --- a/ethosu/vela/tflite/QuantizationParameters.py +++ b/ethosu/vela/tflite/QuantizationParameters.py @@ -10,13 +10,17 @@ class QuantizationParameters(object): __slots__ = ['_tab'] @classmethod - def GetRootAsQuantizationParameters(cls, buf, offset): + def GetRootAs(cls, buf, offset=0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = QuantizationParameters() x.Init(buf, n + offset) return x @classmethod + def GetRootAsQuantizationParameters(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod def QuantizationParametersBufferHasIdentifier(cls, buf, offset, size_prefixed=False): return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed) @@ -157,15 +161,41 @@ class QuantizationParameters(object): return 0 def QuantizationParametersStart(builder): builder.StartObject(7) +def Start(builder): + return QuantizationParametersStart(builder) def QuantizationParametersAddMin(builder, min): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0) +def AddMin(builder, min): + return QuantizationParametersAddMin(builder, min) def QuantizationParametersStartMinVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartMinVector(builder, numElems): + return QuantizationParametersStartMinVector(builder, numElems) def QuantizationParametersAddMax(builder, max): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0) +def AddMax(builder, max): + return QuantizationParametersAddMax(builder, max) def QuantizationParametersStartMaxVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartMaxVector(builder, numElems): + return QuantizationParametersStartMaxVector(builder, numElems) def QuantizationParametersAddScale(builder, scale): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(scale), 0) +def AddScale(builder, scale): + return QuantizationParametersAddScale(builder, scale) def QuantizationParametersStartScaleVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartScaleVector(builder, numElems): + return QuantizationParametersStartScaleVector(builder, numElems) def QuantizationParametersAddZeroPoint(builder, zeroPoint): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(zeroPoint), 0) +def AddZeroPoint(builder, zeroPoint): + return QuantizationParametersAddZeroPoint(builder, zeroPoint) def QuantizationParametersStartZeroPointVector(builder, numElems): return builder.StartVector(8, numElems, 8) +def StartZeroPointVector(builder, numElems): + return QuantizationParametersStartZeroPointVector(builder, numElems) def QuantizationParametersAddDetailsType(builder, detailsType): builder.PrependUint8Slot(4, detailsType, 0) +def AddDetailsType(builder, detailsType): + return QuantizationParametersAddDetailsType(builder, detailsType) def QuantizationParametersAddDetails(builder, details): builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(details), 0) +def AddDetails(builder, details): + return QuantizationParametersAddDetails(builder, details) def QuantizationParametersAddQuantizedDimension(builder, quantizedDimension): builder.PrependInt32Slot(6, quantizedDimension, 0) +def AddQuantizedDimension(builder, quantizedDimension): + return QuantizationParametersAddQuantizedDimension(builder, quantizedDimension) def QuantizationParametersEnd(builder): return builder.EndObject() +def End(builder): + return QuantizationParametersEnd(builder)
\ No newline at end of file |