From c6939a4d269968a34b0ae0aa579f0f0736aaeccc Mon Sep 17 00:00:00 2001 From: Tai Ly Date: Mon, 21 Aug 2023 17:00:29 +0000 Subject: Add is_unranked flag to Tensors This adds a bool field, is_unranked to TosaTensor in tosa.fbs to indicate whether a tensor with shape = {} is an unranked tensor or an 0-D tensor. For older tosa files without this field, the default value is false. Signed-off-by: Tai Ly Change-Id: I86950050b522565509863c483cd3a3c1c50f8f69 --- 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 d8264f2..6613796 100644 --- a/python/tosa/TosaTensor.py +++ b/python/tosa/TosaTensor.py @@ -103,8 +103,15 @@ class TosaTensor(object): return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) return False + # TosaTensor + def IsUnranked(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14)) + if o != 0: + return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) + return False + def TosaTensorStart(builder): - builder.StartObject(5) + builder.StartObject(6) def Start(builder): TosaTensorStart(builder) @@ -151,6 +158,12 @@ def TosaTensorAddVariable(builder, variable): def AddVariable(builder, variable): TosaTensorAddVariable(builder, variable) +def TosaTensorAddIsUnranked(builder, isUnranked): + builder.PrependBoolSlot(5, isUnranked, 0) + +def AddIsUnranked(builder, isUnranked): + TosaTensorAddIsUnranked(builder, isUnranked) + def TosaTensorEnd(builder): return builder.EndObject() -- cgit v1.2.1