aboutsummaryrefslogtreecommitdiff
path: root/python/serializer/tosa_serializer.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/serializer/tosa_serializer.py')
-rw-r--r--python/serializer/tosa_serializer.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py
index 8f70fb0..e8311ce 100644
--- a/python/serializer/tosa_serializer.py
+++ b/python/serializer/tosa_serializer.py
@@ -361,6 +361,14 @@ class TosaSerializerAttribute(TosaSerializerUnion):
self.ints.append((a.AddInput1Zp, input1_zp))
self.ints.append((a.AddOutputZp, output_zp))
+ def FFTAttribute(self, inverse):
+ from tosa import FFTAttribute as a, Attribute
+
+ self.utype = Attribute.Attribute().FFTAttribute
+ self.optFcns = (a.Start, a.End)
+
+ self.bools.append((a.AddInverse, inverse))
+
class TosaSerializerTensor:
def __init__(