Searched refs:_fft_size_for_grad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 171 def _fft_size_for_grad(grad, rank): function
177 size = _math_ops.cast(_fft_size_for_grad(grad, 1), grad.dtype)
184 1. / _math_ops.cast(_fft_size_for_grad(grad, 1), grad.dtype.real_dtype),
191 size = _math_ops.cast(_fft_size_for_grad(grad, 2), grad.dtype)
198 1. / _math_ops.cast(_fft_size_for_grad(grad, 2), grad.dtype.real_dtype),
205 size = _math_ops.cast(_fft_size_for_grad(grad, 3), grad.dtype)
212 1. / _math_ops.cast(_fft_size_for_grad(grad, 3), grad.dtype.real_dtype),
290 _fft_size_for_grad(op.inputs[0], rank), _dtypes.float32)
316 _fft_size_for_grad(grad, rank), _dtypes.float32))
|