Add is_unranked flag to Tensors

This adds a bool field, is_unranked to TosaTensor in tosa.fbs
to indicate whether a tensor with shape = {} is an unranked tensor
or an 0-D tensor.
For older tosa files without this field, the default value is false.

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

1 files changed, 14 insertions, 1 deletions

diff --git a/python/tosa/TosaTensor.py b/python/tosa/TosaTensor.py
index d8264f2..6613796 100644
--- a/python/tosa/TosaTensor.py
+++ b/python/tosa/TosaTensor.py
@@ -103,8 +103,15 @@ class TosaTensor(object):
             return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
         return False
 
+    # TosaTensor
+    def IsUnranked(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
+        if o != 0:
+            return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
+        return False
+
 def TosaTensorStart(builder):
-    builder.StartObject(5)
+    builder.StartObject(6)
 
 def Start(builder):
     TosaTensorStart(builder)
@@ -151,6 +158,12 @@ def TosaTensorAddVariable(builder, variable):
 def AddVariable(builder, variable):
     TosaTensorAddVariable(builder, variable)
 
+def TosaTensorAddIsUnranked(builder, isUnranked):
+    builder.PrependBoolSlot(5, isUnranked, 0)
+
+def AddIsUnranked(builder, isUnranked):
+    TosaTensorAddIsUnranked(builder, isUnranked)
+
 def TosaTensorEnd(builder):
     return builder.EndObject()