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
40 return _array_ops.shape(input_tensor)[-fft_rank:]
53 fft_length = _array_ops.unstack(_array_ops.shape(input_tensor)[-fft_rank:])
55 return _array_ops.stack(fft_length)
92 return _array_ops.pad(input_tensor, outer_paddings + paddings)
97 input_rank = _array_ops.rank(input_tensor)
98 input_fft_shape = _array_ops.shape(input_tensor)[-fft_rank:]
100 outer_paddings = _array_ops.zeros([outer_dims], fft_length.dtype)
103 fft_length = _array_ops.concat([fft_length[:-1],
106 paddings = _array_ops.concat([outer_paddings, fft_paddings], 0)
[all …]
Ddct_ops.py25 from tensorflow.python.ops import array_ops as _array_ops unknown
106 _array_ops.shape(input)[-1])
115 input = _array_ops.pad(input, paddings=padding)
118 or _array_ops.shape(input)[-1])
122 dct1_input = _array_ops.concat([input, input[..., -2:0:-1]], axis=-1)
142 weights = _array_ops.pad(
143 _array_ops.expand_dims(n1, 0), [[0, axis_dim - 1]],
154 weights = _array_ops.pad(
155 _array_ops.expand_dims(n1, 0), [[0, axis_dim - 1]],
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py90 from tensorflow.python.ops import array_ops as _array_ops unknown
286 identity_op = _array_ops.identity(arg, name=name)