blob: 0e8bdc1d872396ad32124baad5bf0b65ec275eb1 [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
6
7class SkipGramOptions(object):
8 __slots__ = ['_tab']
9
10 @classmethod
11 def GetRootAsSkipGramOptions(cls, buf, offset):
12 n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
13 x = SkipGramOptions()
14 x.Init(buf, n + offset)
15 return x
16
17 # SkipGramOptions
18 def Init(self, buf, pos):
19 self._tab = flatbuffers.table.Table(buf, pos)
20
21 # SkipGramOptions
22 def NgramSize(self):
23 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
24 if o != 0:
25 return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
26 return 0
27
28 # SkipGramOptions
29 def MaxSkipSize(self):
30 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
31 if o != 0:
32 return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
33 return 0
34
35 # SkipGramOptions
36 def IncludeAllNgrams(self):
37 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
38 if o != 0:
39 return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
40 return False
41
42def SkipGramOptionsStart(builder): builder.StartObject(3)
43def SkipGramOptionsAddNgramSize(builder, ngramSize): builder.PrependInt32Slot(0, ngramSize, 0)
44def SkipGramOptionsAddMaxSkipSize(builder, maxSkipSize): builder.PrependInt32Slot(1, maxSkipSize, 0)
45def SkipGramOptionsAddIncludeAllNgrams(builder, includeAllNgrams): builder.PrependBoolSlot(2, includeAllNgrams, 0)
46def SkipGramOptionsEnd(builder): return builder.EndObject()