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