diff options
| author | Luke Hutton <luke.hutton@arm.com> | 2023-01-10 14:50:31 +0000 |
|---|---|---|
| committer | Luke Hutton <luke.hutton@arm.com> | 2023-01-24 13:40:17 +0000 |
| commit | 261b7b62b959a6c7312d810d9152069fdff69f3e (patch) | |
| tree | 2be25cefa14cd21379a9fc6f6c499622b6de8bf8 /verif/frameworks/tensor_gen.py | |
| parent | c253e64710f22016894c0e3ac4e9eb76d62cb2f9 (diff) | |
| download | reference_model-261b7b62b959a6c7312d810d9152069fdff69f3e.tar.gz | |
Add RFFT2d to the reference model
Includes:
* RFFT2d reference implementation
* TFLite framework tests
* Basic TOSA tests
* Serialization submodule upgrade with support for FFT/RFFT
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I2a687e9cf87fb62a26160ea52439ba9830bea36e
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, [] |