Searched refs:_irfft_grad_helper (Results 1 – 1 of 1) sorted by relevance
332 def _irfft_grad_helper(rank, rfft_fn): function463 _ops.RegisterGradient("IRFFT")(_irfft_grad_helper(1, rfft))465 _ops.RegisterGradient("IRFFT2D")(_irfft_grad_helper(2, rfft2d))