1# automatically generated by the FlatBuffers compiler, do not modify
2
3# namespace: tflite
4
5import flatbuffers
6
7class SqueezeOptions(object):
8    __slots__ = ['_tab']
9
10    @classmethod
11    def GetRootAsSqueezeOptions(cls, buf, offset):
12        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
13        x = SqueezeOptions()
14        x.Init(buf, n + offset)
15        return x
16
17    # SqueezeOptions
18    def Init(self, buf, pos):
19        self._tab = flatbuffers.table.Table(buf, pos)
20
21    # SqueezeOptions
22    def SqueezeDims(self, j):
23        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
24        if o != 0:
25            a = self._tab.Vector(o)
26            return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
27        return 0
28
29    # SqueezeOptions
30    def SqueezeDimsAsNumpy(self):
31        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
32        if o != 0:
33            return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
34        return 0
35
36    # SqueezeOptions
37    def SqueezeDimsLength(self):
38        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
39        if o != 0:
40            return self._tab.VectorLen(o)
41        return 0
42
43def SqueezeOptionsStart(builder): builder.StartObject(1)
44def SqueezeOptionsAddSqueezeDims(builder, squeezeDims): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(squeezeDims), 0)
45def SqueezeOptionsStartSqueezeDimsVector(builder, numElems): return builder.StartVector(4, numElems, 4)
46def SqueezeOptionsEnd(builder): return builder.EndObject()
47