def _BatchFFT3DGrad(_, grad): size = math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32) return math_ops.batch_ifft3d(grad) * math_ops.complex(size, 0.)