Example #1
0
def _BatchIFFT2DGrad(_, grad):
    rsize = 1. / math_ops.cast(_FFTSizeForGrad(grad, 2), dtypes.float32)
    return math_ops.batch_fft2d(grad) * math_ops.complex(rsize, 0.)
Example #2
0
def _BatchIFFT2DGrad(_, grad):
  rsize = 1. / math_ops.cast(_FFTSizeForGrad(grad, 2), dtypes.float32)
  return math_ops.batch_fft2d(grad) * math_ops.complex(rsize, 0.)