| H A D | fft_generator.cpp | 111 complex_result = fft2d_r2c(in, size0, size1, target, desc); in generate()
116 complex_result = fft2d_c2c(in, size0, size1, sign, target, desc); in generate()
125 complex_result = fft2d_c2c(in, size0, size1, sign, target, desc); in generate()
133 output(x, y, c) = re(complex_result(x, y)); in generate()
136 output(x, y, c) = mux(c, {re(complex_result(x, y)), im(complex_result(x, y))}); in generate()
157 assert(complex_result.defined()); in schedule()
158 complex_result.compute_at(output, Var::outermost()); in schedule()
165 ComplexFunc complex_result; member in __anon065b5c8a0111::FFTGenerator
|