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

1 files changed, 14 insertions, 1 deletions

diff --git a/python/tosa/FFTAttribute.py b/python/tosa/FFTAttribute.py
index 0f22aa7..d1624c2 100644
--- a/python/tosa/FFTAttribute.py
+++ b/python/tosa/FFTAttribute.py
@@ -35,8 +35,15 @@ class FFTAttribute(object):
             return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
         return False
 
+    # FFTAttribute
+    def LocalBound(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        if o != 0:
+            return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
+        return False
+
 def FFTAttributeStart(builder):
-    builder.StartObject(1)
+    builder.StartObject(2)
 
 def Start(builder):
     FFTAttributeStart(builder)
@@ -47,6 +54,12 @@ def FFTAttributeAddInverse(builder, inverse):
 def AddInverse(builder, inverse):
     FFTAttributeAddInverse(builder, inverse)
 
+def FFTAttributeAddLocalBound(builder, localBound):
+    builder.PrependBoolSlot(1, localBound, 0)
+
+def AddLocalBound(builder, localBound):
+    FFTAttributeAddLocalBound(builder, localBound)
+
 def FFTAttributeEnd(builder):
     return builder.EndObject()