diff options
author | Tim Hall <tim.hall@arm.com> | 2021-02-04 21:31:57 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2021-02-05 11:54:48 +0000 |
commit | 42abec185a95ac8d5f2b32d541f587b3cd306e75 (patch) | |
tree | 7c17eb47cb89a37b9f9d82add61385a5a82c40fe /ethosu/vela/tflite/CumsumOptions.py | |
parent | 73e843f76dd71e4ab5e07a7616c2c4806ca6ac25 (diff) | |
download | ethos-u-vela-42abec185a95ac8d5f2b32d541f587b3cd306e75.tar.gz |
MLBEDSW-3771: Updated to TF 2.4 flatbuffer schema
Updated tflite loader and mappings from tensorflow 2.3 to tensorflow 2.4
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I55884000ee139baf639bb0377008e0534f72fe94
Diffstat (limited to 'ethosu/vela/tflite/CumsumOptions.py')
-rw-r--r-- | ethosu/vela/tflite/CumsumOptions.py | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/ethosu/vela/tflite/CumsumOptions.py b/ethosu/vela/tflite/CumsumOptions.py new file mode 100644 index 00000000..33e086ef --- /dev/null +++ b/ethosu/vela/tflite/CumsumOptions.py @@ -0,0 +1,38 @@ +# automatically generated by the FlatBuffers compiler, do not modify + +# namespace: tflite + +import flatbuffers + +class CumsumOptions(object): + __slots__ = ['_tab'] + + @classmethod + def GetRootAsCumsumOptions(cls, buf, offset): + n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) + x = CumsumOptions() + x.Init(buf, n + offset) + return x + + # CumsumOptions + def Init(self, buf, pos): + self._tab = flatbuffers.table.Table(buf, pos) + + # CumsumOptions + def Exclusive(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) + if o != 0: + return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) + return False + + # CumsumOptions + def Reverse(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) + if o != 0: + return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) + return False + +def CumsumOptionsStart(builder): builder.StartObject(2) +def CumsumOptionsAddExclusive(builder, exclusive): builder.PrependBoolSlot(0, exclusive, 0) +def CumsumOptionsAddReverse(builder, reverse): builder.PrependBoolSlot(1, reverse, 0) +def CumsumOptionsEnd(builder): return builder.EndObject() |