From 5e268097917825ddaa00a86ee95a4a6c4f50124b Mon Sep 17 00:00:00 2001 From: Luke Hutton Date: Thu, 12 Jan 2023 22:20:53 +0000 Subject: Add FFT2d and RFFT2d serialization support Signed-off-by: Luke Hutton Change-Id: I79b3684ae1cf24e71ec7098ed7db5be5bc2f6fb2 --- python/serializer/tosa_serializer.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to '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 @@ 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__( -- cgit v1.2.1