Searched refs:irfft_fn (Results 1 – 2 of 2) sorted by relevance
250 def _rfft_grad_helper(rank, irfft_fn): argument326 the_irfft = irfft_fn(grad, fft_length)
553 rfft_fn, irfft_fn = rffts_for_rank[rank]568 self.evaluate(irfft_fn(x, fft_length))