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 /reference_model/include | |
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 'reference_model/include')
-rw-r--r-- | reference_model/include/operators.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/reference_model/include/operators.h b/reference_model/include/operators.h index d2bcf87..d037631 100644 --- a/reference_model/include/operators.h +++ b/reference_model/include/operators.h @@ -88,6 +88,14 @@ extern "C" tosa_tensor_t client_output, const func_ctx_t& func_ctx); + tosa_status_t tosa_run_fft2d(tosa_tensor_t client_input_real, + tosa_tensor_t client_input_imag, + const bool client_inverse, + tosa_tensor_t client_output_real, + const bool client_local_bound, + tosa_tensor_t client_output_imag, + const func_ctx_t& func_ctx); + tosa_status_t tosa_run_fully_connected(tosa_tensor_t client_input, tosa_tensor_t client_weight, tosa_tensor_t client_bias, @@ -112,6 +120,12 @@ extern "C" tosa_tensor_t client_output, const func_ctx_t& func_ctx); + tosa_status_t tosa_run_rfft2d(tosa_tensor_t client_input, + tosa_tensor_t client_output_real, + const bool client_local_bound, + tosa_tensor_t client_output_imag, + const func_ctx_t& func_ctx); + tosa_status_t tosa_run_transpose_conv2d(tosa_tensor_t client_input, tosa_tensor_t client_weight, tosa_tensor_t client_bias, |