blob: 19a59b6492bc608ab970eed14c5fba4a2dc5475f [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
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +02006from flatbuffers.compat import import_numpy
7np = import_numpy()
Tim Hall79d07d22020-04-27 18:20:16 +01008
9class SkipGramOptions(object):
10 __slots__ = ['_tab']
11
12 @classmethod
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020013 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 = SkipGramOptions()
16 x.Init(buf, n + offset)
17 return x
18
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020019 @classmethod
20 def GetRootAsSkipGramOptions(cls, buf, offset=0):
21 """This method is deprecated. Please switch to GetRootAs."""
22 return cls.GetRootAs(buf, offset)
23 @classmethod
24 def SkipGramOptionsBufferHasIdentifier(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 # SkipGramOptions
28 def Init(self, buf, pos):
29 self._tab = flatbuffers.table.Table(buf, pos)
30
31 # SkipGramOptions
32 def NgramSize(self):
33 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
34 if o != 0:
35 return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
36 return 0
37
38 # SkipGramOptions
39 def MaxSkipSize(self):
40 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
41 if o != 0:
42 return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
43 return 0
44
45 # SkipGramOptions
46 def IncludeAllNgrams(self):
47 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
48 if o != 0:
49 return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
50 return False
51
erik.andersson@arm.comdd49a722022-08-10 15:26:48 +020052def SkipGramOptionsStart(builder): builder.StartObject(3)
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020053def Start(builder):
54 return SkipGramOptionsStart(builder)
erik.andersson@arm.comdd49a722022-08-10 15:26:48 +020055def SkipGramOptionsAddNgramSize(builder, ngramSize): builder.PrependInt32Slot(0, ngramSize, 0)
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020056def AddNgramSize(builder, ngramSize):
57 return SkipGramOptionsAddNgramSize(builder, ngramSize)
erik.andersson@arm.comdd49a722022-08-10 15:26:48 +020058def SkipGramOptionsAddMaxSkipSize(builder, maxSkipSize): builder.PrependInt32Slot(1, maxSkipSize, 0)
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020059def AddMaxSkipSize(builder, maxSkipSize):
60 return SkipGramOptionsAddMaxSkipSize(builder, maxSkipSize)
erik.andersson@arm.comdd49a722022-08-10 15:26:48 +020061def SkipGramOptionsAddIncludeAllNgrams(builder, includeAllNgrams): builder.PrependBoolSlot(2, includeAllNgrams, 0)
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020062def AddIncludeAllNgrams(builder, includeAllNgrams):
63 return SkipGramOptionsAddIncludeAllNgrams(builder, includeAllNgrams)
erik.andersson@arm.comdd49a722022-08-10 15:26:48 +020064def SkipGramOptionsEnd(builder): return builder.EndObject()
erik.andersson@arm.com61f05d92022-09-27 12:06:32 +020065def End(builder):
66 return SkipGramOptionsEnd(builder)