diff options
Diffstat (limited to 'python')
-rw-r--r-- | python/tosa/TosaTensor.py | 15 |
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() |