blob: 0e8bdc1d872396ad32124baad5bf0b65ec275eb1 [file] [log] [blame]
# automatically generated by the FlatBuffers compiler, do not modify
# namespace: tflite
import flatbuffers
class SkipGramOptions(object):
__slots__ = ['_tab']
@classmethod
def GetRootAsSkipGramOptions(cls, buf, offset):
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
x = SkipGramOptions()
x.Init(buf, n + offset)
return x
# SkipGramOptions
def Init(self, buf, pos):
self._tab = flatbuffers.table.Table(buf, pos)
# SkipGramOptions
def NgramSize(self):
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
if o != 0:
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
return 0
# SkipGramOptions
def MaxSkipSize(self):
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
if o != 0:
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
return 0
# SkipGramOptions
def IncludeAllNgrams(self):
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
if o != 0:
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
return False
def SkipGramOptionsStart(builder): builder.StartObject(3)
def SkipGramOptionsAddNgramSize(builder, ngramSize): builder.PrependInt32Slot(0, ngramSize, 0)
def SkipGramOptionsAddMaxSkipSize(builder, maxSkipSize): builder.PrependInt32Slot(1, maxSkipSize, 0)
def SkipGramOptionsAddIncludeAllNgrams(builder, includeAllNgrams): builder.PrependBoolSlot(2, includeAllNgrams, 0)
def SkipGramOptionsEnd(builder): return builder.EndObject()