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