| /external/tensorflow/tensorflow/contrib/crf/python/kernel_tests/ |
| crf_test.py | 35 sequence_lengths):
37 inputs[i][tag_indices[i]] for i in range(sequence_lengths))
40 for i in range(sequence_lengths - 1))
61 for sequence_lengths, inputs, tag_indices in zip(sequence_lengths_list,
68 sequence_lengths=array_ops.expand_dims(sequence_lengths, 0),
73 inputs, transition_params, tag_indices, sequence_lengths)
97 for sequence_lengths, inputs, tag_bitmap in zip(
103 sequence_lengths=array_ops.expand_dims(sequence_lengths, 0)
[all...] |
| /external/tensorflow/tensorflow/contrib/crf/python/ops/ |
| crf.py | 24 unary_scores, gold_tags, sequence_lengths)
31 unary_scores, transition_params, sequence_lengths)
38 [unary_scores, sequence_lengths, transition_params, train_op])
74 def crf_sequence_score(inputs, tag_indices, sequence_lengths,
83 sequence_lengths: A [batch_size] vector of true sequence lengths.
97 sequence_scores = array_ops.where(math_ops.less_equal(sequence_lengths, 0),
104 unary_scores = crf_unary_score(tag_indices, sequence_lengths, inputs)
105 binary_scores = crf_binary_score(tag_indices, sequence_lengths,
119 def crf_multitag_sequence_score(inputs, tag_bitmap, sequence_lengths,
137 sequence_lengths: A [batch_size] vector of true sequence lengths
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| rnn_common.py | 182 def select_last_activations(activations, sequence_lengths):
191 sequence_lengths: A `Tensor` with shape `[batch_size]` or `None`.
196 'select_last_activations', values=[activations, sequence_lengths]):
201 if sequence_lengths is None:
202 sequence_lengths = padded_length
205 indices = math_ops.range(batch_size) * padded_length + sequence_lengths - 1
212 def mask_activations_and_labels(activations, labels, sequence_lengths):
213 """Remove entries outside `sequence_lengths` and returned flattened results.
218 sequence_lengths: A `Tensor` of shape `[batch_size]` with the unpadded
222 activations_masked: `logit` values with those beyond `sequence_lengths`
[all...] |
| /external/tensorflow/tensorflow/core/util/ctc/ |
| ctc_beam_search_test.cc | 119 int sequence_lengths[batch_size] = {timesteps}; local
151 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size);
194 int sequence_lengths[batch_size] = {timesteps}; local
201 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size);
264 int sequence_lengths[batch_size] = {timesteps}; local
296 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size);
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | [all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| cudnn_rnn.py | 380 sequence_lengths=None,
394 sequence_lengths: an int32 array representing the variable sequence
401 time-major form. This param is only effective when 'sequence_lengths'
434 inputs, h, c, self.kernel, sequence_lengths, time_major, training)
498 def _forward(self, inputs, h, c, opaque_params, sequence_lengths, time_major,
507 sequence_lengths=sequence_lengths,
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| decoder.py | 115 def finalize(self, outputs, final_state, sequence_lengths):
234 def finalize(self, outputs, final_state, sequence_lengths):
381 def body(time, outputs_ta, state, inputs, finished, sequence_lengths):
390 sequence_lengths: int32 tensor (keeping track of time of finish).
405 array_ops.fill(array_ops.shape(sequence_lengths), time + 1),
406 sequence_lengths)
|
| beam_search_decoder.py | 335 def finalize(self, outputs, final_state, sequence_lengths):
342 sequence_lengths: An `int64` tensor shaped `[batch_size, beam_width]`.
352 del sequence_lengths
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| cudnn_rnn_grad.py | 88 sequence_lengths=op.inputs[4],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_rnn_ops.cc | 637 // Overloaded function to process the sequence_lengths
642 const Tensor** sequence_lengths,
644 TF_RETURN_IF_ERROR(context->input("sequence_lengths", sequence_lengths));
737 const Tensor* sequence_lengths, bool time_major,
746 if (sequence_lengths != nullptr) {
748 sequence_lengths->template flat<int>().data(), model_shapes.batch_size);
800 Tensor* params_backprop, const Tensor* sequence_lengths, bool time_major,
808 if (sequence_lengths != nullptr) {
810 sequence_lengths->template flat<int>().data(), model_shapes.batch_size)
1281 const Tensor* sequence_lengths = nullptr; local
1667 const Tensor* sequence_lengths = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| functional_rnn.py | 179 def _ApplyLengthsToBatch(sequence_lengths, tf_output):
188 array_ops.reshape(sequence_lengths, [-1, 1]), [1, max_time])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| ctc_loss_op_test.py | 228 # batch_size length vector of sequence_lengths
294 sequence_lengths = constant_op.constant([], dtype=dtypes.int32)
303 sess.run(_ctc_loss_v2(labels, inputs, sequence_lengths))
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_test.py | 152 sequence_lengths=lengths,
650 sequence_lengths=lengths,
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
