summaryrefslogtreecommitdiff
path: root/tools/tflitefile_tool/tflite/QuantizationParameters.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/tflitefile_tool/tflite/QuantizationParameters.py')
-rw-r--r--tools/tflitefile_tool/tflite/QuantizationParameters.py36
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()