Searched refs:_rfft_grad_helper (Results 1 – 1 of 1) sorted by relevance
217 def _rfft_grad_helper(rank, irfft_fn): function328 _ops.RegisterGradient("RFFT")(_rfft_grad_helper(1, irfft))330 _ops.RegisterGradient("RFFT2D")(_rfft_grad_helper(2, irfft2d))