diff options
Diffstat (limited to 'tools/tflitefile_tool/tflite/QuantizationParameters.py')
-rw-r--r-- | tools/tflitefile_tool/tflite/QuantizationParameters.py | 36 |
1 files changed, 31 insertions, 5 deletions
diff --git a/tools/tflitefile_tool/tflite/QuantizationParameters.py b/tools/tflitefile_tool/tflite/QuantizationParameters.py index 7d5e53072..9a752d7e5 100644 --- a/tools/tflitefile_tool/tflite/QuantizationParameters.py +++ b/tools/tflitefile_tool/tflite/QuantizationParameters.py @@ -115,23 +115,40 @@ class QuantizationParameters(object): return self._tab.VectorLen(o) return 0 + # QuantizationParameters + def DetailsType(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) + if o != 0: + return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos) + return 0 + + # QuantizationParameters + def Details(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14)) + if o != 0: + from flatbuffers.table import Table + obj = Table(bytearray(), 0) + self._tab.Union(obj, o) + return obj + return None + def QuantizationParametersStart(builder): - builder.StartObject(4) + builder.StartObject(6) -def QuantizationParametersAddMin(builder, min): +def QuantizationParametersAddMin(builder, minSlot): builder.PrependUOffsetTRelativeSlot( - 0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0) + 0, flatbuffers.number_types.UOffsetTFlags.py_type(minSlot), 0) def QuantizationParametersStartMinVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def QuantizationParametersAddMax(builder, max): +def QuantizationParametersAddMax(builder, maxSlot): builder.PrependUOffsetTRelativeSlot( - 1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0) + 1, flatbuffers.number_types.UOffsetTFlags.py_type(maxSlot), 0) def QuantizationParametersStartMaxVector(builder, numElems): @@ -156,5 +173,14 @@ 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 QuantizationParametersEnd(builder): return builder.EndObject() |