Searched refs:rfft_fn (Results 1 – 2 of 2) sorted by relevance
332 def _irfft_grad_helper(rank, rfft_fn): argument366 the_rfft = rfft_fn(grad, fft_length)
553 rfft_fn, irfft_fn = rffts_for_rank[rank]560 self.evaluate(rfft_fn(x, fft_length))