Searched refs:seq_lengths (Results 1 - 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | reverse_sequence_op_test.py | 37 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 D | reverse_sequence_op_test.py | 35 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 D | reverse_sequence_op.h | 33 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 D | rnn.py | 429 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 D | array_grad.py | 617 seq_lengths = op.inputs[1] 623 seq_lengths=seq_lengths), None
|
H A D | array_ops.py | 2636 seq_lengths, 2648 seq_lengths=seq_lengths,
|
/external/tensorflow/tensorflow/cc/gradients/ |
H A D | array_grad.cc | 227 auto seq_lengths = op.input(1); local 233 ReverseSequence(scope, grad_inputs[0], seq_lengths, seq_dim,
|
H A D | array_grad_test.cc | 203 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 D | lstm_ops_test.py | 366 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