From 4881c29247d4b411de446b13d9bd58ea93737aac Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Wed, 1 Nov 2023 16:12:07 -0700 Subject: Add support for local_bound attribute local_bound is used to determine when fast convolution algorithms can be used in implementing the operation. Signed-off-by: Eric Kunze Change-Id: I9970a2544e90a620f46ac4d3d01cec90a15710a9 --- python/tosa/TransposeConvAttribute.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'python/tosa/TransposeConvAttribute.py') diff --git a/python/tosa/TransposeConvAttribute.py b/python/tosa/TransposeConvAttribute.py index def507e..a74a433 100644 --- a/python/tosa/TransposeConvAttribute.py +++ b/python/tosa/TransposeConvAttribute.py @@ -123,8 +123,15 @@ class TransposeConvAttribute(object): return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos) return 0 + # TransposeConvAttribute + def LocalBound(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 TransposeConvAttributeStart(builder): - builder.StartObject(5) + builder.StartObject(6) def Start(builder): TransposeConvAttributeStart(builder) @@ -177,6 +184,12 @@ def TransposeConvAttributeAddWeightZp(builder, weightZp): def AddWeightZp(builder, weightZp): TransposeConvAttributeAddWeightZp(builder, weightZp) +def TransposeConvAttributeAddLocalBound(builder, localBound): + builder.PrependBoolSlot(5, localBound, 0) + +def AddLocalBound(builder, localBound): + TransposeConvAttributeAddLocalBound(builder, localBound) + def TransposeConvAttributeEnd(builder): return builder.EndObject() -- cgit v1.2.1