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