Searched refs:_fft_size_for_grad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 204 def _fft_size_for_grad(grad, rank): function
210 size = _math_ops.cast(_fft_size_for_grad(grad, 1), grad.dtype)
217 1. / _math_ops.cast(_fft_size_for_grad(grad, 1), grad.dtype.real_dtype),
224 size = _math_ops.cast(_fft_size_for_grad(grad, 2), grad.dtype)
231 1. / _math_ops.cast(_fft_size_for_grad(grad, 2), grad.dtype.real_dtype),
238 size = _math_ops.cast(_fft_size_for_grad(grad, 3), grad.dtype)
245 1. / _math_ops.cast(_fft_size_for_grad(grad, 3), grad.dtype.real_dtype),
325 _fft_size_for_grad(op.inputs[0], rank), real_dtype)
360 _fft_size_for_grad(grad, rank), real_dtype))
|