tools/tflitefile_tool/tflite/QuantizationParameters.py

   1 # automatically generated by the FlatBuffers compiler, do not modify
   2
   3 # namespace: onert_tflite
   4
   5 import flatbuffers
   6
   7
   8 class QuantizationParameters(object):
   9     __slots__ = ['_tab']
  10
  11     @classmethod
  12     def GetRootAsQuantizationParameters(cls, buf, offset):
  13         n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
  14         x = QuantizationParameters()
  15         x.Init(buf, n + offset)
  16         return x
  17
  18     # QuantizationParameters
  19     def Init(self, buf, pos):
  20         self._tab = flatbuffers.table.Table(buf, pos)
  21
  22     # QuantizationParameters
  23     def Min(self, j):
  24         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
  25         if o != 0:
  26             a = self._tab.Vector(o)
  27             return self._tab.Get(
  28                 flatbuffers.number_types.Float32Flags,
  29                 a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
  30         return 0
  31
  32     # QuantizationParameters
  33     def MinAsNumpy(self):
  34         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
  35         if o != 0:
  36             return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
  37         return 0
  38
  39     # QuantizationParameters
  40     def MinLength(self):
  41         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
  42         if o != 0:
  43             return self._tab.VectorLen(o)
  44         return 0
  45
  46     # QuantizationParameters
  47     def Max(self, j):
  48         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
  49         if o != 0:
  50             a = self._tab.Vector(o)
  51             return self._tab.Get(
  52                 flatbuffers.number_types.Float32Flags,
  53                 a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
  54         return 0
  55
  56     # QuantizationParameters
  57     def MaxAsNumpy(self):
  58         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
  59         if o != 0:
  60             return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
  61         return 0
  62
  63     # QuantizationParameters
  64     def MaxLength(self):
  65         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
  66         if o != 0:
  67             return self._tab.VectorLen(o)
  68         return 0
  69
  70     # QuantizationParameters
  71     def Scale(self, j):
  72         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
  73         if o != 0:
  74             a = self._tab.Vector(o)
  75             return self._tab.Get(
  76                 flatbuffers.number_types.Float32Flags,
  77                 a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
  78         return 0
  79
  80     # QuantizationParameters
  81     def ScaleAsNumpy(self):
  82         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
  83         if o != 0:
  84             return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
  85         return 0
  86
  87     # QuantizationParameters
  88     def ScaleLength(self):
  89         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
  90         if o != 0:
  91             return self._tab.VectorLen(o)
  92         return 0
  93
  94     # QuantizationParameters
  95     def ZeroPoint(self, j):
  96         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
  97         if o != 0:
  98             a = self._tab.Vector(o)
  99             return self._tab.Get(
 100                 flatbuffers.number_types.Int64Flags,
 101                 a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8))
 102         return 0
 103
 104     # QuantizationParameters
 105     def ZeroPointAsNumpy(self):
 106         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
 107         if o != 0:
 108             return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o)
 109         return 0
 110
 111     # QuantizationParameters
 112     def ZeroPointLength(self):
 113         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
 114         if o != 0:
 115             return self._tab.VectorLen(o)
 116         return 0
 117
 118     # QuantizationParameters
 119     def DetailsType(self):
 120         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
 121         if o != 0:
 122             return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
 123         return 0
 124
 125     # QuantizationParameters
 126     def Details(self):
 127         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
 128         if o != 0:
 129             from flatbuffers.table import Table
 130             obj = Table(bytearray(), 0)
 131             self._tab.Union(obj, o)
 132             return obj
 133         return None
 134
 135     # QuantizationParameters
 136     def QuantizedDimension(self):
 137         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
 138         if o != 0:
 139             return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
 140         return 0
 141
 142
 143 def QuantizationParametersStart(builder):
 144     builder.StartObject(7)
 145
 146
 147 def QuantizationParametersAddMin(builder, min):
 148     builder.PrependUOffsetTRelativeSlot(
 149         0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0)
 150
 151
 152 def QuantizationParametersStartMinVector(builder, numElems):
 153     return builder.StartVector(4, numElems, 4)
 154
 155
 156 def QuantizationParametersAddMax(builder, max):
 157     builder.PrependUOffsetTRelativeSlot(
 158         1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0)
 159
 160
 161 def QuantizationParametersStartMaxVector(builder, numElems):
 162     return builder.StartVector(4, numElems, 4)
 163
 164
 165 def QuantizationParametersAddScale(builder, scale):
 166     builder.PrependUOffsetTRelativeSlot(
 167         2, flatbuffers.number_types.UOffsetTFlags.py_type(scale), 0)
 168
 169
 170 def QuantizationParametersStartScaleVector(builder, numElems):
 171     return builder.StartVector(4, numElems, 4)
 172
 173
 174 def QuantizationParametersAddZeroPoint(builder, zeroPoint):
 175     builder.PrependUOffsetTRelativeSlot(
 176         3, flatbuffers.number_types.UOffsetTFlags.py_type(zeroPoint), 0)
 177
 178
 179 def QuantizationParametersStartZeroPointVector(builder, numElems):
 180     return builder.StartVector(8, numElems, 8)
 181
 182
 183 def QuantizationParametersAddDetailsType(builder, detailsType):
 184     builder.PrependUint8Slot(4, detailsType, 0)
 185
 186
 187 def QuantizationParametersAddDetails(builder, details):
 188     builder.PrependUOffsetTRelativeSlot(
 189         5, flatbuffers.number_types.UOffsetTFlags.py_type(details), 0)
 190
 191
 192 def QuantizationParametersAddQuantizedDimension(builder, quantizedDimension):
 193     builder.PrependInt32Slot(6, quantizedDimension, 0)
 194
 195
 196 def QuantizationParametersEnd(builder):
 197     return builder.EndObject()