diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-01-12 22:20:53 +0000 |
---|---|---|
committer | Luke Hutton <luke.hutton@arm.com> | 2023-01-13 10:34:25 +0000 |
commit | 5e268097917825ddaa00a86ee95a4a6c4f50124b (patch) | |
tree | 8cc0f3b0b7a1d8e68e192a4e243020dc69791c5f /python/serializer/tosa_serializer.py | |
parent | ca7ce0e94b3ee7339f31b47baa3a3fb4522243a2 (diff) | |
download | serialization_lib-5e268097917825ddaa00a86ee95a4a6c4f50124b.tar.gz |
Add FFT2d and RFFT2d serialization support
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I79b3684ae1cf24e71ec7098ed7db5be5bc2f6fb2
Diffstat (limited to 'python/serializer/tosa_serializer.py')
-rw-r--r-- | python/serializer/tosa_serializer.py | 8 |
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__( |