diff options
Diffstat (limited to 'verif/frameworks/tensor_gen.py')
-rw-r--r-- | verif/frameworks/tensor_gen.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/verif/frameworks/tensor_gen.py b/verif/frameworks/tensor_gen.py index 767989e..c534a58 100644 --- a/verif/frameworks/tensor_gen.py +++ b/verif/frameworks/tensor_gen.py @@ -274,3 +274,12 @@ class TGen: ) return tf_placeholders, tf_consts + + @staticmethod + def tgRFFT2d(op, shape, dtype, rng): + # Require rank 3 shape + if len(shape) != 3: + return [], [] + + tf_placeholders = [("placeholder_0", TGen.getRand(shape, dtype, rng))] + return tf_placeholders, [] |