blob: 0a27af0e9912930510ab95c2187aa23868757915 [file] [log] [blame]
Tim Hall79d07d22020-04-27 18:20:16 +01001# automatically generated by the FlatBuffers compiler, do not modify
2
3# namespace: tflite
4
5import flatbuffers
Dwight Lidman8a12da12021-07-19 13:43:05 +02006from flatbuffers.compat import import_numpy
7np = import_numpy()
Tim Hall79d07d22020-04-27 18:20:16 +01008
9class QuantizationParameters(object):
10 __slots__ = ['_tab']
11
12 @classmethod
Rickard Bolin2de898a2021-12-20 08:35:23 +000013 def GetRootAs(cls, buf, offset=0):
Tim Hall79d07d22020-04-27 18:20:16 +010014 n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
15 x = QuantizationParameters()
16 x.Init(buf, n + offset)
17 return x
18
Dwight Lidman8a12da12021-07-19 13:43:05 +020019 @classmethod
Rickard Bolin2de898a2021-12-20 08:35:23 +000020 def GetRootAsQuantizationParameters(cls, buf, offset=0):
21 """This method is deprecated. Please switch to GetRootAs."""
22 return cls.GetRootAs(buf, offset)
23 @classmethod
Dwight Lidman8a12da12021-07-19 13:43:05 +020024 def QuantizationParametersBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
25 return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
26
Tim Hall79d07d22020-04-27 18:20:16 +010027 # QuantizationParameters
28 def Init(self, buf, pos):
29 self._tab = flatbuffers.table.Table(buf, pos)
30
31 # QuantizationParameters
32 def Min(self, j):
33 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
34 if o != 0:
35 a = self._tab.Vector(o)
36 return self._tab.Get(flatbuffers.number_types.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
37 return 0
38
39 # QuantizationParameters
40 def MinAsNumpy(self):
41 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
42 if o != 0:
43 return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
44 return 0
45
46 # QuantizationParameters
47 def MinLength(self):
48 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
49 if o != 0:
50 return self._tab.VectorLen(o)
51 return 0
52
53 # QuantizationParameters
Dwight Lidman8a12da12021-07-19 13:43:05 +020054 def MinIsNone(self):
55 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
56 return o == 0
57
58 # QuantizationParameters
Tim Hall79d07d22020-04-27 18:20:16 +010059 def Max(self, j):
60 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
61 if o != 0:
62 a = self._tab.Vector(o)
63 return self._tab.Get(flatbuffers.number_types.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
64 return 0
65
66 # QuantizationParameters
67 def MaxAsNumpy(self):
68 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
69 if o != 0:
70 return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
71 return 0
72
73 # QuantizationParameters
74 def MaxLength(self):
75 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
76 if o != 0:
77 return self._tab.VectorLen(o)
78 return 0
79
80 # QuantizationParameters
Dwight Lidman8a12da12021-07-19 13:43:05 +020081 def MaxIsNone(self):
82 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
83 return o == 0
84
85 # QuantizationParameters
Tim Hall79d07d22020-04-27 18:20:16 +010086 def Scale(self, j):
87 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
88 if o != 0:
89 a = self._tab.Vector(o)
90 return self._tab.Get(flatbuffers.number_types.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
91 return 0
92
93 # QuantizationParameters
94 def ScaleAsNumpy(self):
95 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
96 if o != 0:
97 return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
98 return 0
99
100 # QuantizationParameters
101 def ScaleLength(self):
102 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
103 if o != 0:
104 return self._tab.VectorLen(o)
105 return 0
106
107 # QuantizationParameters
Dwight Lidman8a12da12021-07-19 13:43:05 +0200108 def ScaleIsNone(self):
109 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
110 return o == 0
111
112 # QuantizationParameters
Tim Hall79d07d22020-04-27 18:20:16 +0100113 def ZeroPoint(self, j):
114 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
115 if o != 0:
116 a = self._tab.Vector(o)
117 return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8))
118 return 0
119
120 # QuantizationParameters
121 def ZeroPointAsNumpy(self):
122 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
123 if o != 0:
124 return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o)
125 return 0
126
127 # QuantizationParameters
128 def ZeroPointLength(self):
129 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
130 if o != 0:
131 return self._tab.VectorLen(o)
132 return 0
133
134 # QuantizationParameters
Dwight Lidman8a12da12021-07-19 13:43:05 +0200135 def ZeroPointIsNone(self):
136 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
137 return o == 0
138
139 # QuantizationParameters
Tim Hall79d07d22020-04-27 18:20:16 +0100140 def DetailsType(self):
141 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
142 if o != 0:
143 return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
144 return 0
145
146 # QuantizationParameters
147 def Details(self):
148 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
149 if o != 0:
150 from flatbuffers.table import Table
151 obj = Table(bytearray(), 0)
152 self._tab.Union(obj, o)
153 return obj
154 return None
155
156 # QuantizationParameters
157 def QuantizedDimension(self):
158 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
159 if o != 0:
160 return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
161 return 0
162
163def QuantizationParametersStart(builder): builder.StartObject(7)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000164def Start(builder):
165 return QuantizationParametersStart(builder)
Tim Hall79d07d22020-04-27 18:20:16 +0100166def QuantizationParametersAddMin(builder, min): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000167def AddMin(builder, min):
168 return QuantizationParametersAddMin(builder, min)
Tim Hall79d07d22020-04-27 18:20:16 +0100169def QuantizationParametersStartMinVector(builder, numElems): return builder.StartVector(4, numElems, 4)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000170def StartMinVector(builder, numElems):
171 return QuantizationParametersStartMinVector(builder, numElems)
Tim Hall79d07d22020-04-27 18:20:16 +0100172def QuantizationParametersAddMax(builder, max): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000173def AddMax(builder, max):
174 return QuantizationParametersAddMax(builder, max)
Tim Hall79d07d22020-04-27 18:20:16 +0100175def QuantizationParametersStartMaxVector(builder, numElems): return builder.StartVector(4, numElems, 4)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000176def StartMaxVector(builder, numElems):
177 return QuantizationParametersStartMaxVector(builder, numElems)
Tim Hall79d07d22020-04-27 18:20:16 +0100178def QuantizationParametersAddScale(builder, scale): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(scale), 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000179def AddScale(builder, scale):
180 return QuantizationParametersAddScale(builder, scale)
Tim Hall79d07d22020-04-27 18:20:16 +0100181def QuantizationParametersStartScaleVector(builder, numElems): return builder.StartVector(4, numElems, 4)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000182def StartScaleVector(builder, numElems):
183 return QuantizationParametersStartScaleVector(builder, numElems)
Tim Hall79d07d22020-04-27 18:20:16 +0100184def QuantizationParametersAddZeroPoint(builder, zeroPoint): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(zeroPoint), 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000185def AddZeroPoint(builder, zeroPoint):
186 return QuantizationParametersAddZeroPoint(builder, zeroPoint)
Tim Hall79d07d22020-04-27 18:20:16 +0100187def QuantizationParametersStartZeroPointVector(builder, numElems): return builder.StartVector(8, numElems, 8)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000188def StartZeroPointVector(builder, numElems):
189 return QuantizationParametersStartZeroPointVector(builder, numElems)
Tim Hall79d07d22020-04-27 18:20:16 +0100190def QuantizationParametersAddDetailsType(builder, detailsType): builder.PrependUint8Slot(4, detailsType, 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000191def AddDetailsType(builder, detailsType):
192 return QuantizationParametersAddDetailsType(builder, detailsType)
Tim Hall79d07d22020-04-27 18:20:16 +0100193def QuantizationParametersAddDetails(builder, details): builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(details), 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000194def AddDetails(builder, details):
195 return QuantizationParametersAddDetails(builder, details)
Tim Hall79d07d22020-04-27 18:20:16 +0100196def QuantizationParametersAddQuantizedDimension(builder, quantizedDimension): builder.PrependInt32Slot(6, quantizedDimension, 0)
Rickard Bolin2de898a2021-12-20 08:35:23 +0000197def AddQuantizedDimension(builder, quantizedDimension):
198 return QuantizationParametersAddQuantizedDimension(builder, quantizedDimension)
Tim Hall79d07d22020-04-27 18:20:16 +0100199def QuantizationParametersEnd(builder): return builder.EndObject()
Rickard Bolin2de898a2021-12-20 08:35:23 +0000200def End(builder):
201 return QuantizationParametersEnd(builder)