def _tf_fn(x): return spectral_ops.irfft2d( x, fft_length=[x.shape[-2].value, 2 * (x.shape[-1].value - 1)])