Searched refs:irfft_fn (Results 1 – 2 of 2) sorted by relevance
217 def _rfft_grad_helper(rank, irfft_fn): argument291 the_irfft = irfft_fn(grad, fft_length)
510 rfft_fn, irfft_fn = rffts_for_rank[rank]525 self.evaluate(irfft_fn(x, fft_length))