From 442261bf67fa2ec4d86ed3e431a6373787b3e35a Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Fri, 9 Sep 2022 13:38:56 -0700 Subject: [Serialization_lib] Support StatefulOps for TOSA - Add variable in TosaTensor to schema file - Update TosaSerializationTensor regarding variable change - Rename internal zero_pad() and expose interface as ForceAlignTensorData() Signed-off-by: Jerry Ge Change-Id: I36fa64eb0802cb5b8d3564ea7233460ef8c9f539 --- python/tosa/TosaTensor.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'python/tosa/TosaTensor.py') 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() -- cgit v1.2.1