diff options
Diffstat (limited to 'src/xnnpack/fft.h')
| -rw-r--r-- | src/xnnpack/fft.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/xnnpack/fft.h b/src/xnnpack/fft.h index 5cb9369e0..1b673bafd 100644 --- a/src/xnnpack/fft.h +++ b/src/xnnpack/fft.h @@ -27,6 +27,18 @@ DECLARE_CS16_BFLY4_UKERNEL_FUNCTION(xnn_cs16_bfly4_ukernel__scalar_x2) DECLARE_CS16_BFLY4_UKERNEL_FUNCTION(xnn_cs16_bfly4_ukernel__scalar_x3) DECLARE_CS16_BFLY4_UKERNEL_FUNCTION(xnn_cs16_bfly4_ukernel__scalar_x4) +#define DECLARE_CS16_FFTR_UKERNEL_FUNCTION(fn_name) \ + XNN_INTERNAL void fn_name( \ + size_t samples, \ + const int16_t* input, \ + int16_t* output, \ + const int16_t* twiddle); + +DECLARE_CS16_FFTR_UKERNEL_FUNCTION(xnn_cs16_fftr_ukernel__scalar_x1) +DECLARE_CS16_FFTR_UKERNEL_FUNCTION(xnn_cs16_fftr_ukernel__scalar_x2) +DECLARE_CS16_FFTR_UKERNEL_FUNCTION(xnn_cs16_fftr_ukernel__scalar_x3) +DECLARE_CS16_FFTR_UKERNEL_FUNCTION(xnn_cs16_fftr_ukernel__scalar_x4) + #ifdef __cplusplus } // extern "C" #endif |