From 35a3aa994cf18f735193a05a7eb2c61d497233d2 Mon Sep 17 00:00:00 2001 From: Dhruv Chauhan Date: Tue, 28 Nov 2023 15:00:34 +0000 Subject: 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 --- reference_model/include/operators.h | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'reference_model/include') 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, -- cgit v1.2.1