Searched refs:seq_lengths (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
H A Dreverse_sequence_op_test.py37 seq_lengths,
43 x, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=seq_lengths)
56 seq_lengths,
59 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth, True,
61 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth,
73 seq_lengths = np.asarray([3, 0, 4], dtype=len_dtype)
87 self._testBothReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
121 seq_lengths = np.asarray([3, 0, 4], dtype=np.int64)
125 seq_lengths_t = constant_op.constant(seq_lengths, shap
[all...]
/external/tensorflow/tensorflow/compiler/tests/
H A Dreverse_sequence_op_test.py35 seq_lengths,
40 lengths = array_ops.placeholder(dtypes.as_dtype(seq_lengths.dtype))
43 p, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=lengths)
45 tf_ans = ans.eval(feed_dict={p: x, lengths: seq_lengths})
49 ans.eval(feed_dict={p: x, lengths: seq_lengths})
58 seq_lengths=np.array([1, 3, 2], np.int32),
70 seq_lengths = np.asarray([3, 0, 4], dtype=len_dtype)
84 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
/external/tensorflow/tensorflow/core/kernels/
H A Dreverse_sequence_op.h33 int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths)
37 seq_lengths_(seq_lengths) {}
66 typename TTypes<Tlen>::ConstVec seq_lengths,
69 seq_dim, seq_lengths);
32 ReverseGenerator(typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) argument
63 Compute( const Device& d, typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths, typename TTypes<T, Dims>::Tensor output) argument
/external/tensorflow/tensorflow/python/ops/
H A Drnn.py429 def _reverse(input_, seq_lengths, seq_dim, batch_dim):
430 if seq_lengths is not None:
432 input=input_, seq_lengths=seq_lengths,
439 inputs, seq_lengths=sequence_length,
448 tmp, seq_lengths=sequence_length,
H A Darray_grad.py617 seq_lengths = op.inputs[1]
623 seq_lengths=seq_lengths), None
H A Darray_ops.py2636 seq_lengths,
2648 seq_lengths=seq_lengths,
/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc227 auto seq_lengths = op.input(1); local
233 ReverseSequence(scope, grad_inputs[0], seq_lengths, seq_dim,
H A Darray_grad_test.cc203 auto seq_lengths = Const(scope_, {1, 2, 3, 4, 5}); local
205 auto y = ReverseSequence(scope_, x, seq_lengths, /* seq_dim */ 2);
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
H A Dlstm_ops_test.py366 seq_lengths = constant_op.constant([3, 4, 5])
390 cell_inputs, dtype=dtypes.float32, sequence_length=seq_lengths)
404 lengths = [int(i < l) for l in seq_lengths.eval()]

Completed in 418 milliseconds