diff options
author | Tai Ly <tai.ly@arm.com> | 2023-09-19 21:30:18 +0000 |
---|---|---|
committer | Tai Ly <tai.ly@arm.com> | 2023-09-19 21:55:11 +0000 |
commit | d0520b9b3a0eaf9dadc6cdb57ed42906e577d32e (patch) | |
tree | 1538ac688cbafa4e8e08b7aabe3e10bbe404c767 /python | |
parent | 924f3094a745c1955d51fce18b488adfed5ee76b (diff) | |
download | serialization_lib-d0520b9b3a0eaf9dadc6cdb57ed42906e577d32e.tar.gz |
Add variable_name to tensors
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Ia142c8b1a9e0869daefb3eef71100fd0c2a0effc
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 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() |