Add variable_name to tensors

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Ia142c8b1a9e0869daefb3eef71100fd0c2a0effc

1 files changed, 14 insertions, 1 deletions

diff --git a/python/tosa/TosaTensor.py b/python/tosa/TosaTensor.py
index 6613796..3fb9f86 100644
--- a/python/tosa/TosaTensor.py
+++ b/python/tosa/TosaTensor.py
@@ -110,8 +110,15 @@ class TosaTensor(object):
             return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
         return False
 
+    # TosaTensor
+    def VariableName(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
+        if o != 0:
+            return self._tab.String(o + self._tab.Pos)
+        return None
+
 def TosaTensorStart(builder):
-    builder.StartObject(6)
+    builder.StartObject(7)
 
 def Start(builder):
     TosaTensorStart(builder)
@@ -164,6 +171,12 @@ def TosaTensorAddIsUnranked(builder, isUnranked):
 def AddIsUnranked(builder, isUnranked):
     TosaTensorAddIsUnranked(builder, isUnranked)
 
+def TosaTensorAddVariableName(builder, variableName):
+    builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(variableName), 0)
+
+def AddVariableName(builder, variableName):
+    TosaTensorAddVariableName(builder, variableName)
+
 def TosaTensorEnd(builder):
     return builder.EndObject()