Home
last modified time | relevance | path

Searched refs:fft_input_output (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Drfft2d.cc247 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 Dirfft2d.cc230 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 …]