Home
last modified time | relevance | path

Searched refs:_array_ops (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py25 from tensorflow.python.ops import array_ops as _array_ops unknown
38 return _array_ops.shape(input_tensor)[-fft_rank:]
51 fft_length = _array_ops.unstack(_array_ops.shape(input_tensor)[-fft_rank:])
53 return _array_ops.stack(fft_length)
90 return _array_ops.pad(input_tensor, outer_paddings + paddings)
95 input_rank = _array_ops.rank(input_tensor)
96 input_fft_shape = _array_ops.shape(input_tensor)[-fft_rank:]
98 outer_paddings = _array_ops.zeros([outer_dims], fft_length.dtype)
101 fft_length = _array_ops.concat([fft_length[:-1],
104 paddings = _array_ops.concat([outer_paddings, fft_paddings], 0)
[all …]
Ddct_ops.py25 from tensorflow.python.ops import array_ops as _array_ops unknown
96 or _array_ops.shape(input)[-1])
100 dct1_input = _array_ops.concat([input, input[..., -2:0:-1]], axis=-1)
120 weights = _array_ops.pad(
121 _array_ops.expand_dims(n1, 0), [[0, axis_dim - 1]],
132 weights = _array_ops.pad(
133 _array_ops.expand_dims(n1, 0), [[0, axis_dim - 1]],
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py85 from tensorflow.python.ops import array_ops as _array_ops unknown
274 identity_op = _array_ops.identity(arg, name=name)