diff options
| author | Dhruv Chauhan <dhruv.chauhan@arm.com> | 2023-11-28 15:00:34 +0000 |
|---|---|---|
| committer | Dhruv Chauhan <dhruv.chauhan@arm.com> | 2023-11-28 17:21:01 +0000 |
| commit | 35a3aa994cf18f735193a05a7eb2c61d497233d2 (patch) | |
| tree | 2add19eb59682b9d1c04c2ac30e67d2ab36a998e /scripts | |
| parent | a015001dfbd0ed48caf54fd66b0509ee344a229e (diff) | |
| download | reference_model-35a3aa994cf18f735193a05a7eb2c61d497233d2.tar.gz | |
Fix Fast Fourier Transforms in operator API
* Change ignore list in generate_api.py to generate operators
information.
* Fix serialization attributes mapping for operator FFT and RFFT
* Add a unit test for Fft2d and Rfft2d operator
Change-Id: I3ad7a77a3c46aa586834188bab42cbdcc423e834
Signed-off-by: Dhruv Chauhan <dhruv.chauhan@arm.com>
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/operator_api/generate_api.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scripts/operator_api/generate_api.py b/scripts/operator_api/generate_api.py index d9077f0..7f10568 100644 --- a/scripts/operator_api/generate_api.py +++ b/scripts/operator_api/generate_api.py @@ -62,6 +62,8 @@ def getSerializeOpType(tosaOpName): "conv3d": "Conv", "depthwise_conv2d": "Conv", "fully_connected": "FullyConnected", + "fft2d": "FFT", + "rfft2d": "RFFT", "matmul": "MatMul", "max_pool2d": "Pool", "transpose_conv2d": "TransposeConv", @@ -236,8 +238,6 @@ def getOperators(tosaXml): "cond_if", "const", "custom", - "fft2d", - "rfft2d", "variable", "variable_read", "variable_write", |