Home
last modified time | relevance | path

Searched refs:_math_ops (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py28 from tensorflow.python.ops import math_ops as _math_ops unknown
54 fft_length[-1] = _math_ops.maximum(0, 2 * (fft_length[-1] - 1))
99 outer_dims = _math_ops.maximum(0, input_rank - fft_rank)
105 fft_paddings = _math_ops.maximum(0, fft_length - input_fft_shape)
205 return _math_ops.reduce_prod(_array_ops.shape(grad)[-rank:])
210 size = _math_ops.cast(_fft_size_for_grad(grad, 1), grad.dtype)
216 rsize = _math_ops.cast(
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)
230 rsize = _math_ops.cast(
[all …]
Ddct_ops.py22 from tensorflow.python.ops import math_ops as _math_ops unknown
115 axis_dim_float = _math_ops.cast(axis_dim, input.dtype)
119 dct1 = _math_ops.real(fft_ops.rfft(dct1_input))
123 scale = 2.0 * _math_ops.exp(
124 _math_ops.complex(
125 zero, -_math_ops.range(axis_dim_float) * _math.pi * 0.5 /
130 dct2 = _math_ops.real(
135 n1 = 0.5 * _math_ops.rsqrt(axis_dim_float)
147 n1 = _math_ops.sqrt(axis_dim_float)
156 scale = 2.0 * _math_ops.exp(
[all …]
/external/tensorflow/tensorflow/python/framework/experimental/
Dmath_ops.py17 from tensorflow.python.framework.experimental import _math_ops
23 return _math_ops.add(ctx, a, b, name)
28 return _math_ops.mat_mul(ctx, a, b, name)
33 return _math_ops.neg(ctx, a, name)
38 return _math_ops.sub(ctx, a, b, name)
43 return _math_ops.mul(ctx, a, b, name)
48 return _math_ops.log1p(ctx, a, name)
53 return _math_ops.div_no_nan(ctx, a, b, name)
DBUILD45 name = "_math_ops",
88 ":_math_ops",
Dmath_ops.cc32 PYBIND11_MODULE(_math_ops, m) { in PYBIND11_MODULE() argument