diff options
Diffstat (limited to 'python')
-rw-r--r-- | python/serializer/tosa_serializer.py | 20 | ||||
-rw-r--r-- | python/tosa/ConvAttribute.py | 22 | ||||
-rw-r--r-- | python/tosa/PoolAttribute.py | 22 | ||||
-rw-r--r-- | python/tosa/ReshapeAttribute.py | 22 | ||||
-rw-r--r-- | python/tosa/SliceAttribute.py | 22 | ||||
-rw-r--r-- | python/tosa/TableAttribute.py | 6 | ||||
-rw-r--r-- | python/tosa/TransposeAttribute.py | 22 |
7 files changed, 68 insertions, 68 deletions
diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py index 660619b..fec676e 100644 --- a/python/serializer/tosa_serializer.py +++ b/python/serializer/tosa_serializer.py @@ -137,23 +137,23 @@ class TosaSerializerAttribute(TosaSerializerUnion): def __init__(self): super().__init__() - def PoolAttribute(self, kernel, stride, padding): + def PoolAttribute(self, kernel, stride, pad): from tosa import PoolAttribute as a, Attribute self.utype = Attribute.Attribute().PoolAttribute self.optFcns = (a.Start, a.End) - self.intvecs.append((a.AddPadding, padding)) + self.intvecs.append((a.AddPad, pad)) self.intvecs.append((a.AddKernel, kernel)) self.intvecs.append((a.AddStride, stride)) - def ConvAttribute(self, padding, stride, dilation): + def ConvAttribute(self, pad, stride, dilation): from tosa import ConvAttribute as a, Attribute self.utype = Attribute.Attribute().ConvAttribute self.optFcns = (a.Start, a.End) - self.intvecs.append((a.AddPadding, padding)) + self.intvecs.append((a.AddPad, pad)) self.intvecs.append((a.AddStride, stride)) self.intvecs.append((a.AddDilation, dilation)) @@ -186,21 +186,21 @@ class TosaSerializerAttribute(TosaSerializerUnion): self.ints.append((a.AddAxis, axis)) - def ReshapeAttribute(self, shape): + def ReshapeAttribute(self, new_shape): from tosa import ReshapeAttribute as a, Attribute self.utype = Attribute.Attribute().ReshapeAttribute self.optFcns = (a.Start, a.End) - self.intvecs.append((a.AddShape, shape)) + self.intvecs.append((a.AddNewShape, new_shape)) - def SliceAttribute(self, begin, size): + def SliceAttribute(self, start, size): from tosa import SliceAttribute as a, Attribute self.utype = Attribute.Attribute().SliceAttribute self.optFcns = (a.Start, a.End) - self.intvecs.append((a.AddBegin, begin)) + self.intvecs.append((a.AddStart, start)) self.intvecs.append((a.AddSize, size)) def TileAttribute(self, multiples): @@ -292,13 +292,13 @@ class TosaSerializerAttribute(TosaSerializerUnion): self.strings.append((a.AddCondBranch, cond_branch)) self.strings.append((a.AddBodyBranch, body_branch)) - def TransposeAttribute(self, perm): + def TransposeAttribute(self, perms): from tosa import TransposeAttribute as a, Attribute self.utype = Attribute.Attribute().TransposeAttribute self.optFcns = (a.Start, a.End) - self.intvecs.append((a.AddPerm, perm)) + self.intvecs.append((a.AddPerms, perms)) def TableAttribute(self, table): from tosa import TableAttribute as a, Attribute diff --git a/python/tosa/ConvAttribute.py b/python/tosa/ConvAttribute.py index bbb5300..2d69163 100644 --- a/python/tosa/ConvAttribute.py +++ b/python/tosa/ConvAttribute.py @@ -29,7 +29,7 @@ class ConvAttribute(object): self._tab = flatbuffers.table.Table(buf, pos) # ConvAttribute - def Padding(self, j): + def Pad(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) @@ -37,21 +37,21 @@ class ConvAttribute(object): return 0 # ConvAttribute - def PaddingAsNumpy(self): + def PadAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) return 0 # ConvAttribute - def PaddingLength(self): + def PadLength(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.VectorLen(o) return 0 # ConvAttribute - def PaddingIsNone(self): + def PadIsNone(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) return o == 0 @@ -113,14 +113,14 @@ def Start(builder): builder.StartObject(3) def ConvAttributeStart(builder): """This method is deprecated. Please switch to Start.""" return Start(builder) -def AddPadding(builder, padding): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(padding), 0) -def ConvAttributeAddPadding(builder, padding): - """This method is deprecated. Please switch to AddPadding.""" - return AddPadding(builder, padding) -def StartPaddingVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def ConvAttributeStartPaddingVector(builder, numElems): +def AddPad(builder, pad): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(pad), 0) +def ConvAttributeAddPad(builder, pad): + """This method is deprecated. Please switch to AddPad.""" + return AddPad(builder, pad) +def StartPadVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def ConvAttributeStartPadVector(builder, numElems): """This method is deprecated. Please switch to Start.""" - return StartPaddingVector(builder, numElems) + return StartPadVector(builder, numElems) def AddStride(builder, stride): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(stride), 0) def ConvAttributeAddStride(builder, stride): """This method is deprecated. Please switch to AddStride.""" diff --git a/python/tosa/PoolAttribute.py b/python/tosa/PoolAttribute.py index 47a3173..0399d00 100644 --- a/python/tosa/PoolAttribute.py +++ b/python/tosa/PoolAttribute.py @@ -29,7 +29,7 @@ class PoolAttribute(object): self._tab = flatbuffers.table.Table(buf, pos) # PoolAttribute - def Padding(self, j): + def Pad(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) @@ -37,21 +37,21 @@ class PoolAttribute(object): return 0 # PoolAttribute - def PaddingAsNumpy(self): + def PadAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) return 0 # PoolAttribute - def PaddingLength(self): + def PadLength(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.VectorLen(o) return 0 # PoolAttribute - def PaddingIsNone(self): + def PadIsNone(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) return o == 0 @@ -113,14 +113,14 @@ def Start(builder): builder.StartObject(3) def PoolAttributeStart(builder): """This method is deprecated. Please switch to Start.""" return Start(builder) -def AddPadding(builder, padding): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(padding), 0) -def PoolAttributeAddPadding(builder, padding): - """This method is deprecated. Please switch to AddPadding.""" - return AddPadding(builder, padding) -def StartPaddingVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def PoolAttributeStartPaddingVector(builder, numElems): +def AddPad(builder, pad): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(pad), 0) +def PoolAttributeAddPad(builder, pad): + """This method is deprecated. Please switch to AddPad.""" + return AddPad(builder, pad) +def StartPadVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def PoolAttributeStartPadVector(builder, numElems): """This method is deprecated. Please switch to Start.""" - return StartPaddingVector(builder, numElems) + return StartPadVector(builder, numElems) def AddKernel(builder, kernel): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(kernel), 0) def PoolAttributeAddKernel(builder, kernel): """This method is deprecated. Please switch to AddKernel.""" diff --git a/python/tosa/ReshapeAttribute.py b/python/tosa/ReshapeAttribute.py index d0f9135..0d335d8 100644 --- a/python/tosa/ReshapeAttribute.py +++ b/python/tosa/ReshapeAttribute.py @@ -29,7 +29,7 @@ class ReshapeAttribute(object): self._tab = flatbuffers.table.Table(buf, pos) # ReshapeAttribute - def Shape(self, j): + def NewShape(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) @@ -37,21 +37,21 @@ class ReshapeAttribute(object): return 0 # ReshapeAttribute - def ShapeAsNumpy(self): + def NewShapeAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) return 0 # ReshapeAttribute - def ShapeLength(self): + def NewShapeLength(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.VectorLen(o) return 0 # ReshapeAttribute - def ShapeIsNone(self): + def NewShapeIsNone(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) return o == 0 @@ -59,14 +59,14 @@ def Start(builder): builder.StartObject(1) def ReshapeAttributeStart(builder): """This method is deprecated. Please switch to Start.""" return Start(builder) -def AddShape(builder, shape): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0) -def ReshapeAttributeAddShape(builder, shape): - """This method is deprecated. Please switch to AddShape.""" - return AddShape(builder, shape) -def StartShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def ReshapeAttributeStartShapeVector(builder, numElems): +def AddNewShape(builder, newShape): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(newShape), 0) +def ReshapeAttributeAddNewShape(builder, newShape): + """This method is deprecated. Please switch to AddNewShape.""" + return AddNewShape(builder, newShape) +def StartNewShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def ReshapeAttributeStartNewShapeVector(builder, numElems): """This method is deprecated. Please switch to Start.""" - return StartShapeVector(builder, numElems) + return StartNewShapeVector(builder, numElems) def End(builder): return builder.EndObject() def ReshapeAttributeEnd(builder): """This method is deprecated. Please switch to End.""" diff --git a/python/tosa/SliceAttribute.py b/python/tosa/SliceAttribute.py index d7d476f..37bc666 100644 --- a/python/tosa/SliceAttribute.py +++ b/python/tosa/SliceAttribute.py @@ -29,7 +29,7 @@ class SliceAttribute(object): self._tab = flatbuffers.table.Table(buf, pos) # SliceAttribute - def Begin(self, j): + def Start(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) @@ -37,21 +37,21 @@ class SliceAttribute(object): return 0 # SliceAttribute - def BeginAsNumpy(self): + def StartAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) return 0 # SliceAttribute - def BeginLength(self): + def StartLength(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.VectorLen(o) return 0 # SliceAttribute - def BeginIsNone(self): + def StartIsNone(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) return o == 0 @@ -86,14 +86,14 @@ def Start(builder): builder.StartObject(2) def SliceAttributeStart(builder): """This method is deprecated. Please switch to Start.""" return Start(builder) -def AddBegin(builder, begin): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(begin), 0) -def SliceAttributeAddBegin(builder, begin): - """This method is deprecated. Please switch to AddBegin.""" - return AddBegin(builder, begin) -def StartBeginVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def SliceAttributeStartBeginVector(builder, numElems): +def AddStart(builder, start): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(start), 0) +def SliceAttributeAddStart(builder, start): + """This method is deprecated. Please switch to AddStart.""" + return AddStart(builder, start) +def StartStartVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def SliceAttributeStartStartVector(builder, numElems): """This method is deprecated. Please switch to Start.""" - return StartBeginVector(builder, numElems) + return StartStartVector(builder, numElems) def AddSize(builder, size): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(size), 0) def SliceAttributeAddSize(builder, size): """This method is deprecated. Please switch to AddSize.""" diff --git a/python/tosa/TableAttribute.py b/python/tosa/TableAttribute.py index 87990c5..6fd719f 100644 --- a/python/tosa/TableAttribute.py +++ b/python/tosa/TableAttribute.py @@ -33,14 +33,14 @@ class TableAttribute(object): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) - return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4)) + return self._tab.Get(flatbuffers.number_types.Int16Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 2)) return 0 # TableAttribute def TableAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: - return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) + return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int16Flags, o) return 0 # TableAttribute @@ -63,7 +63,7 @@ def AddTable(builder, table): builder.PrependUOffsetTRelativeSlot(0, flatbuffers def TableAttributeAddTable(builder, table): """This method is deprecated. Please switch to AddTable.""" return AddTable(builder, table) -def StartTableVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartTableVector(builder, numElems): return builder.StartVector(2, numElems, 2) def TableAttributeStartTableVector(builder, numElems): """This method is deprecated. Please switch to Start.""" return StartTableVector(builder, numElems) diff --git a/python/tosa/TransposeAttribute.py b/python/tosa/TransposeAttribute.py index 4c9b034..1e42493 100644 --- a/python/tosa/TransposeAttribute.py +++ b/python/tosa/TransposeAttribute.py @@ -29,7 +29,7 @@ class TransposeAttribute(object): self._tab = flatbuffers.table.Table(buf, pos) # TransposeAttribute - def Perm(self, j): + def Perms(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) @@ -37,21 +37,21 @@ class TransposeAttribute(object): return 0 # TransposeAttribute - def PermAsNumpy(self): + def PermsAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) return 0 # TransposeAttribute - def PermLength(self): + def PermsLength(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.VectorLen(o) return 0 # TransposeAttribute - def PermIsNone(self): + def PermsIsNone(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) return o == 0 @@ -59,14 +59,14 @@ def Start(builder): builder.StartObject(1) def TransposeAttributeStart(builder): """This method is deprecated. Please switch to Start.""" return Start(builder) -def AddPerm(builder, perm): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(perm), 0) -def TransposeAttributeAddPerm(builder, perm): - """This method is deprecated. Please switch to AddPerm.""" - return AddPerm(builder, perm) -def StartPermVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def TransposeAttributeStartPermVector(builder, numElems): +def AddPerms(builder, perms): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(perms), 0) +def TransposeAttributeAddPerms(builder, perms): + """This method is deprecated. Please switch to AddPerms.""" + return AddPerms(builder, perms) +def StartPermsVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def TransposeAttributeStartPermsVector(builder, numElems): """This method is deprecated. Please switch to Start.""" - return StartPermVector(builder, numElems) + return StartPermsVector(builder, numElems) def End(builder): return builder.EndObject() def TransposeAttributeEnd(builder): """This method is deprecated. Please switch to End.""" |