[Serialization_lib] Support StatefulOps for TOSA

- Add variable in TosaTensor to schema file
- Update TosaSerializationTensor regarding variable change
- Rename internal zero_pad() and expose interface as ForceAlignTensorData()

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I36fa64eb0802cb5b8d3564ea7233460ef8c9f539

1 files changed, 14 insertions, 1 deletions

diff --git a/python/tosa/TosaTensor.py b/python/tosa/TosaTensor.py
index 850ff8f..d8264f2 100644
--- a/python/tosa/TosaTensor.py
+++ b/python/tosa/TosaTensor.py
@@ -96,8 +96,15 @@ class TosaTensor(object):
         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
         return o == 0
 
+    # TosaTensor
+    def Variable(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
+        if o != 0:
+            return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
+        return False
+
 def TosaTensorStart(builder):
-    builder.StartObject(4)
+    builder.StartObject(5)
 
 def Start(builder):
     TosaTensorStart(builder)
@@ -138,6 +145,12 @@ def TosaTensorStartDataVector(builder, numElems):
 def StartDataVector(builder, numElems: int) -> int:
     return TosaTensorStartDataVector(builder, numElems)
 
+def TosaTensorAddVariable(builder, variable):
+    builder.PrependBoolSlot(4, variable, 0)
+
+def AddVariable(builder, variable):
+    TosaTensorAddVariable(builder, variable)
+
 def TosaTensorEnd(builder):
     return builder.EndObject()