1 files changed, 28 insertions, 4 deletions
diff --git a/ethosu/vela/tosa/AxisAttribute.py b/ethosu/vela/tosa/AxisAttribute.py
index 77f1bca..7ce4a63 100644
--- a/ethosu/vela/tosa/AxisAttribute.py
+++ b/ethosu/vela/tosa/AxisAttribute.py
@@ -3,17 +3,27 @@
 # namespace: tosa
 
 import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
 
 class AxisAttribute(object):
     __slots__ = ['_tab']
 
     @classmethod
-    def GetRootAsAxisAttribute(cls, buf, offset):
+    def GetRootAs(cls, buf, offset=0):
         n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
         x = AxisAttribute()
         x.Init(buf, n + offset)
         return x
 
+    @classmethod
+    def GetRootAsAxisAttribute(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    @classmethod
+    def AxisAttributeBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+        return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed)
+
     # AxisAttribute
     def Init(self, buf, pos):
         self._tab = flatbuffers.table.Table(buf, pos)
@@ -25,6 +35,20 @@ class AxisAttribute(object):
             return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
         return 0
 
-def AxisAttributeStart(builder): builder.StartObject(1)
-def AxisAttributeAddAxis(builder, axis): builder.PrependInt32Slot(0, axis, 0)
-def AxisAttributeEnd(builder): return builder.EndObject()
+def AxisAttributeStart(builder):
+    builder.StartObject(1)
+
+def Start(builder):
+    AxisAttributeStart(builder)
+
+def AxisAttributeAddAxis(builder, axis):
+    builder.PrependInt32Slot(0, axis, 0)
+
+def AddAxis(builder, axis):
+    AxisAttributeAddAxis(builder, axis)
+
+def AxisAttributeEnd(builder):
+    return builder.EndObject()
+
+def End(builder):
+    return AxisAttributeEnd(builder)