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