1 files changed, 8 insertions, 0 deletions
diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index 8870f41..6e7b6a5 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -1243,3 +1243,11 @@ class TBuilder:
 
         def eval(self, a):
             return self.dense(a)
+
+    class RFFT2d:
+        def __init__(self, fft_length, name):
+            self.fft_length = fft_length
+            self.result_name = name
+
+        def eval(self, a):
+            return tf.signal.rfft2d(a, self.fft_length, name=self.result_name)