commit 5e268097917825ddaa00a86ee95a4a6c4f50124b
author Luke Hutton <luke.hutton@arm.com> Thu Jan 12 22:20:53 2023 +0000
committer Luke Hutton <luke.hutton@arm.com> Fri Jan 13 10:34:25 2023 +0000
tree 8cc0f3b0b7a1d8e68e192a4e243020dc69791c5f
parent ca7ce0e94b3ee7339f31b47baa3a3fb4522243a2

Add FFT2d and RFFT2d serialization support

Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I79b3684ae1cf24e71ec7098ed7db5be5bc2f6fb2

python/serializer/tosa_serializer.py

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 @@
         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__(