diff options
Diffstat (limited to 'ethosu/vela/tflite/QuantizationParameters.py')
-rw-r--r-- | ethosu/vela/tflite/QuantizationParameters.py | 97 |
1 files changed, 14 insertions, 83 deletions
diff --git a/ethosu/vela/tflite/QuantizationParameters.py b/ethosu/vela/tflite/QuantizationParameters.py index 8b6a35e5..fcd686cf 100644 --- a/ethosu/vela/tflite/QuantizationParameters.py +++ b/ethosu/vela/tflite/QuantizationParameters.py @@ -3,27 +3,17 @@ # namespace: tflite import flatbuffers -from flatbuffers.compat import import_numpy -np = import_numpy() class QuantizationParameters(object): __slots__ = ['_tab'] @classmethod - def GetRootAs(cls, buf, offset=0): + def GetRootAsQuantizationParameters(cls, buf, offset): 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) - # QuantizationParameters def Init(self, buf, pos): self._tab = flatbuffers.table.Table(buf, pos) @@ -51,11 +41,6 @@ class QuantizationParameters(object): return 0 # QuantizationParameters - def MinIsNone(self): - o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) - return o == 0 - - # QuantizationParameters def Max(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) if o != 0: @@ -78,11 +63,6 @@ class QuantizationParameters(object): return 0 # QuantizationParameters - def MaxIsNone(self): - o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) - return o == 0 - - # QuantizationParameters def Scale(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) if o != 0: @@ -105,11 +85,6 @@ class QuantizationParameters(object): return 0 # QuantizationParameters - def ScaleIsNone(self): - o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) - return o == 0 - - # QuantizationParameters def ZeroPoint(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) if o != 0: @@ -132,11 +107,6 @@ class QuantizationParameters(object): return 0 # QuantizationParameters - def ZeroPointIsNone(self): - o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) - return o == 0 - - # QuantizationParameters def DetailsType(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) if o != 0: @@ -160,55 +130,16 @@ class QuantizationParameters(object): return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos) return 0 -def Start(builder): builder.StartObject(7) -def QuantizationParametersStart(builder): - """This method is deprecated. Please switch to Start.""" - return Start(builder) -def AddMin(builder, min): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0) -def QuantizationParametersAddMin(builder, min): - """This method is deprecated. Please switch to AddMin.""" - return AddMin(builder, min) -def StartMinVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def QuantizationParametersStartMinVector(builder, numElems): - """This method is deprecated. Please switch to Start.""" - return StartMinVector(builder, numElems) -def AddMax(builder, max): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0) -def QuantizationParametersAddMax(builder, max): - """This method is deprecated. Please switch to AddMax.""" - return AddMax(builder, max) -def StartMaxVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def QuantizationParametersStartMaxVector(builder, numElems): - """This method is deprecated. Please switch to Start.""" - return StartMaxVector(builder, numElems) -def AddScale(builder, scale): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(scale), 0) -def QuantizationParametersAddScale(builder, scale): - """This method is deprecated. Please switch to AddScale.""" - return AddScale(builder, scale) -def StartScaleVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def QuantizationParametersStartScaleVector(builder, numElems): - """This method is deprecated. Please switch to Start.""" - return StartScaleVector(builder, numElems) -def AddZeroPoint(builder, zeroPoint): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(zeroPoint), 0) -def QuantizationParametersAddZeroPoint(builder, zeroPoint): - """This method is deprecated. Please switch to AddZeroPoint.""" - return AddZeroPoint(builder, zeroPoint) -def StartZeroPointVector(builder, numElems): return builder.StartVector(8, numElems, 8) -def QuantizationParametersStartZeroPointVector(builder, numElems): - """This method is deprecated. Please switch to Start.""" - return StartZeroPointVector(builder, numElems) -def AddDetailsType(builder, detailsType): builder.PrependUint8Slot(4, detailsType, 0) -def QuantizationParametersAddDetailsType(builder, detailsType): - """This method is deprecated. Please switch to AddDetailsType.""" - return AddDetailsType(builder, detailsType) -def AddDetails(builder, details): builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(details), 0) -def QuantizationParametersAddDetails(builder, details): - """This method is deprecated. Please switch to AddDetails.""" - return AddDetails(builder, details) -def AddQuantizedDimension(builder, quantizedDimension): builder.PrependInt32Slot(6, quantizedDimension, 0) -def QuantizationParametersAddQuantizedDimension(builder, quantizedDimension): - """This method is deprecated. Please switch to AddQuantizedDimension.""" - return AddQuantizedDimension(builder, quantizedDimension) -def End(builder): return builder.EndObject() -def QuantizationParametersEnd(builder): - """This method is deprecated. Please switch to End.""" - return End(builder)
\ No newline at end of file +def QuantizationParametersStart(builder): builder.StartObject(7) +def QuantizationParametersAddMin(builder, min): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0) +def QuantizationParametersStartMinVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def QuantizationParametersAddMax(builder, max): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0) +def QuantizationParametersStartMaxVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def QuantizationParametersAddScale(builder, scale): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(scale), 0) +def QuantizationParametersStartScaleVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def QuantizationParametersAddZeroPoint(builder, zeroPoint): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(zeroPoint), 0) +def QuantizationParametersStartZeroPointVector(builder, numElems): return builder.StartVector(8, numElems, 8) +def QuantizationParametersAddDetailsType(builder, detailsType): builder.PrependUint8Slot(4, detailsType, 0) +def QuantizationParametersAddDetails(builder, details): builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(details), 0) +def QuantizationParametersAddQuantizedDimension(builder, quantizedDimension): builder.PrependInt32Slot(6, quantizedDimension, 0) +def QuantizationParametersEnd(builder): return builder.EndObject() |