blob: 5aa23e2caa1e64822ffc1f9cd7630bd6edd1ff2a [file] [log] [blame]
Kevin Cheng38d214c2021-10-15 15:49:19 -07001# automatically generated by the FlatBuffers compiler, do not modify
2
Kevin Cheng38d214c2021-10-15 15:49:19 -07003# namespace: tosa
4
5import flatbuffers
6from flatbuffers.compat import import_numpy
7np = import_numpy()
8
9class TransposeAttribute(object):
10 __slots__ = ['_tab']
11
12 @classmethod
Eric Kunze4381b3d2022-08-22 18:15:41 +000013 def GetRootAs(cls, buf, offset=0):
Kevin Cheng38d214c2021-10-15 15:49:19 -070014 n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
15 x = TransposeAttribute()
16 x.Init(buf, n + offset)
17 return x
18
19 @classmethod
Eric Kunze4381b3d2022-08-22 18:15:41 +000020 def GetRootAsTransposeAttribute(cls, buf, offset=0):
21 """This method is deprecated. Please switch to GetRootAs."""
22 return cls.GetRootAs(buf, offset)
23 @classmethod
Kevin Cheng38d214c2021-10-15 15:49:19 -070024 def TransposeAttributeBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
25 return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed)
26
27 # TransposeAttribute
28 def Init(self, buf, pos):
29 self._tab = flatbuffers.table.Table(buf, pos)
30
31 # TransposeAttribute
TatWai Chong7be71652022-05-10 17:26:20 -070032 def Perms(self, j):
Kevin Cheng38d214c2021-10-15 15:49:19 -070033 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
34 if o != 0:
35 a = self._tab.Vector(o)
36 return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
37 return 0
38
39 # TransposeAttribute
TatWai Chong7be71652022-05-10 17:26:20 -070040 def PermsAsNumpy(self):
Kevin Cheng38d214c2021-10-15 15:49:19 -070041 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
42 if o != 0:
43 return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
44 return 0
45
46 # TransposeAttribute
TatWai Chong7be71652022-05-10 17:26:20 -070047 def PermsLength(self):
Kevin Cheng38d214c2021-10-15 15:49:19 -070048 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
49 if o != 0:
50 return self._tab.VectorLen(o)
51 return 0
52
53 # TransposeAttribute
TatWai Chong7be71652022-05-10 17:26:20 -070054 def PermsIsNone(self):
Kevin Cheng38d214c2021-10-15 15:49:19 -070055 o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
56 return o == 0
57
Tai Ly89963aa2023-07-03 22:14:05 +000058def TransposeAttributeStart(builder):
59 builder.StartObject(1)
60
Eric Kunze4381b3d2022-08-22 18:15:41 +000061def Start(builder):
Tai Ly89963aa2023-07-03 22:14:05 +000062 TransposeAttributeStart(builder)
63
64def TransposeAttributeAddPerms(builder, perms):
65 builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(perms), 0)
66
Eric Kunze4381b3d2022-08-22 18:15:41 +000067def AddPerms(builder, perms):
Tai Ly89963aa2023-07-03 22:14:05 +000068 TransposeAttributeAddPerms(builder, perms)
69
70def TransposeAttributeStartPermsVector(builder, numElems):
71 return builder.StartVector(4, numElems, 4)
72
Tai Ly3aebe2b2024-04-24 16:03:22 +000073def StartPermsVector(builder, numElems):
Eric Kunze4381b3d2022-08-22 18:15:41 +000074 return TransposeAttributeStartPermsVector(builder, numElems)
Tai Ly89963aa2023-07-03 22:14:05 +000075
76def TransposeAttributeEnd(builder):
77 return builder.EndObject()
78
Eric Kunze4381b3d2022-08-22 18:15:41 +000079def End(builder):
Tai Ly89963aa2023-07-03 22:14:05 +000080 return TransposeAttributeEnd(builder)