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 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() |