Searched refs:_math_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 27 from tensorflow.python.ops import math_ops as _math_ops unknown
52 fft_length[-1] = _math_ops.maximum(0, 2 * (fft_length[-1] - 1))
97 outer_dims = _math_ops.maximum(0, input_rank - fft_rank)
103 fft_paddings = _math_ops.maximum(0, fft_length - input_fft_shape)
172 return _math_ops.reduce_prod(_array_ops.shape(grad)[-rank:])
177 size = _math_ops.cast(_fft_size_for_grad(grad, 1), grad.dtype)
183 rsize = _math_ops.cast(
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)
197 rsize = _math_ops.cast(
[all …]
|
| D | dct_ops.py | 26 from tensorflow.python.ops import math_ops as _math_ops unknown
97 axis_dim_float = _math_ops.cast(axis_dim, _dtypes.float32)
101 dct1 = _math_ops.real(fft_ops.rfft(dct1_input))
105 scale = 2.0 * _math_ops.exp(
106 _math_ops.complex(
107 0.0, -_math_ops.range(axis_dim_float) * _math.pi * 0.5 /
112 dct2 = _math_ops.real(
117 n1 = 0.5 * _math_ops.rsqrt(axis_dim_float)
118 n2 = n1 * _math_ops.sqrt(2.0)
129 n1 = _math_ops.sqrt(axis_dim_float)
[all …]
|