Searched refs:_rfft_wrapper (Results 1 – 1 of 1) sorted by relevance
112 def _rfft_wrapper(fft_fn, fft_rank, default_name): function184 rfft = _rfft_wrapper(gen_spectral_ops.rfft, 1, "rfft")190 rfft2d = _rfft_wrapper(gen_spectral_ops.rfft2d, 2, "rfft2d")196 rfft3d = _rfft_wrapper(gen_spectral_ops.rfft3d, 3, "rfft3d")