From d0520b9b3a0eaf9dadc6cdb57ed42906e577d32e Mon Sep 17 00:00:00 2001 From: Tai Ly Date: Tue, 19 Sep 2023 21:30:18 +0000 Subject: Add variable_name to tensors Signed-off-by: Tai Ly Change-Id: Ia142c8b1a9e0869daefb3eef71100fd0c2a0effc --- python/tosa/TosaTensor.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'python/tosa') 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() -- cgit v1.2.1