Searched refs:fft_input_output (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | rfft2d.cc | 247 void Rfft2dReorder(int fft_height, int fft_width, double** fft_input_output) { in Rfft2dReorder() argument 264 real = fft_input_output[i][0]; in Rfft2dReorder() 265 img = fft_input_output[i][1]; in Rfft2dReorder() 266 fft_input_output[i][fft_width] = img; in Rfft2dReorder() 267 fft_input_output[i][fft_width + 1] = real; in Rfft2dReorder() 268 fft_input_output[fft_height - i][fft_width] = img; in Rfft2dReorder() 269 fft_input_output[fft_height - i][fft_width + 1] = -real; in Rfft2dReorder() 270 fft_input_output[i][0] = fft_input_output[fft_height - i][0]; in Rfft2dReorder() 271 fft_input_output[i][1] = -fft_input_output[fft_height - i][1]; in Rfft2dReorder() 274 double temp = fft_input_output[0][1]; in Rfft2dReorder() [all …]
|
H A D | irfft2d.cc | 230 void Irfft2dReorder(int fft_height, int fft_width, double** fft_input_output) { in Irfft2dReorder() argument 235 fft_input_output[i][j] = -fft_input_output[i][j]; in Irfft2dReorder() 241 rdft2dsort(fft_height, fft_width, kBackwardFft, fft_input_output); in Irfft2dReorder() 244 void Irfft2dImpl(int fft_height, int fft_width, double** fft_input_output, in Irfft2dImpl() argument 248 Irfft2dReorder(fft_height, fft_width, fft_input_output); in Irfft2dImpl() 253 rdft2d(fft_height, fft_width, kBackwardFft, fft_input_output, in Irfft2dImpl() 260 double** fft_input_output) { in PrepareInputBuffer() argument 266 fft_input_output[i][2 * j] = input_data[in_pos].real(); in PrepareInputBuffer() 267 fft_input_output[i][2 * j + 1] = input_data[in_pos].imag(); in PrepareInputBuffer() 272 fft_input_output[i][2 * j] = 0; in PrepareInputBuffer() [all …]
|