aboutsummaryrefslogtreecommitdiff
path: root/python/tosa/TransposeConvAttribute.py
diff options
context:
space:
mode:
authorEric Kunze <eric.kunze@arm.com>2023-11-01 16:12:07 -0700
committerEric Kunze <eric.kunze@arm.com>2023-11-28 14:44:41 -0800
commitdee09f349c538d5a9752d742163a474f051c627a (patch)
tree5f2dbea65968a7270d53e7e633d24f3054b21633 /python/tosa/TransposeConvAttribute.py
parent38887b919ac3152e6e2735c68b915cff352df3bf (diff)
downloadserialization_lib-dee09f349c538d5a9752d742163a474f051c627a.tar.gz
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 <eric.kunze@arm.com> Change-Id: I9970a2544e90a620f46ac4d3d01cec90a15710a9
Diffstat (limited to 'python/tosa/TransposeConvAttribute.py')
-rw-r--r--python/tosa/TransposeConvAttribute.py15
1 files changed, 14 insertions, 1 deletions
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()