Searched refs:rfft_fn (Results 1 – 2 of 2) sorted by relevance
297 def _irfft_grad_helper(rank, rfft_fn): argument322 the_rfft = rfft_fn(grad, fft_length)
510 rfft_fn, irfft_fn = rffts_for_rank[rank]517 self.evaluate(rfft_fn(x, fft_length))