| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 480 sequence_lengths): argument
500 'sequence_lengths': sequence_lengths,
515 'sequence_lengths': sequence_lengths
544 time_major, go_backwards, sequence_lengths, argument
621 input_length=sequence_lengths
622 if sequence_lengths is not None else timesteps,
628 go_backwards, sequence_lengths): argument
664 sequence_lengths = calculate_sequence_by_mask(mask, time_major)
666 if sequence_lengths is not None:
674 inputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | sequence_feature_column.py | 151 sequence_lengths = []
163 sequence_lengths.append(sequence_length)
166 fc._verify_static_batch_size_equality(sequence_lengths,
168 sequence_length = _assert_all_equal_and_return(sequence_lengths)
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search_test.cc | 123 int sequence_lengths[batch_size] = {timesteps}; in ctc_beam_search_decoding_with_and_without_dictionary() local
157 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size); in ctc_beam_search_decoding_with_and_without_dictionary()
208 int sequence_lengths[batch_size] = {timesteps}; in ctc_beam_search_decoding_all_beam_elements_have_finite_scores() local
215 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size); in ctc_beam_search_decoding_all_beam_elements_have_finite_scores()
285 int sequence_lengths[batch_size] = {timesteps}; in ctc_beam_search_label_selection() local
320 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size); in ctc_beam_search_label_selection()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CudnnRNNV3.pbtxt | 20 name: "sequence_lengths"
145 name: "sequence_lengths"
277 name: "sequence_lengths"
|
| D | CudnnRNNBackpropV3.pbtxt | 20 name: "sequence_lengths"
166 name: "sequence_lengths"
319 name: "sequence_lengths"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 683 const Tensor** sequence_lengths, const int num_proj, in ExtractForwardInput() argument
685 TF_RETURN_IF_ERROR(context->input("sequence_lengths", sequence_lengths)); in ExtractForwardInput()
795 const Tensor* sequence_lengths, bool time_major, in DoForward() argument
805 if (sequence_lengths != nullptr) { in DoForward()
807 sequence_lengths->template flat<int>().data(), model_shapes.batch_size); in DoForward()
859 Tensor* params_backprop, const Tensor* sequence_lengths, bool time_major, in DoBackward() argument
868 if (sequence_lengths != nullptr) { in DoBackward()
870 sequence_lengths->template flat<int>().data(), model_shapes.batch_size); in DoBackward()
948 bool ShouldUsePaddedIO(const Tensor* sequence_lengths, in ShouldUsePaddedIO() argument
951 auto seq_array = sequence_lengths->template flat<int>().data(); in ShouldUsePaddedIO()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CudnnRNNV3.pbtxt | 20 name: "sequence_lengths"
145 name: "sequence_lengths"
|
| D | CudnnRNNBackpropV3.pbtxt | 20 name: "sequence_lengths"
166 name: "sequence_lengths"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNV3.pbtxt | 7 buffer. Accepts one extra input "sequence_lengths" than CudnnRNN.
31 sequence_lengths: a vector of lengths of each input sequence.
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 7 "sequence_lengths" input than CudnnRNNBackprop.
31 sequence_lengths: a vector of lengths of each input sequence.
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 88 sequence_lengths=op.inputs[4],
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | ctc_loss_op_test.py | 297 sequence_lengths = constant_op.constant([], dtype=dtypes.int32)
306 sess.run(_ctc_loss_v2(labels, inputs, sequence_lengths))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 957 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'output\',…
985 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'rnn_mode\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 957 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'output\',…
985 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'rnn_mode\…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 5502 … tf.Output, input_h tf.Output, input_c tf.Output, params tf.Output, sequence_lengths tf.Output, ou…
5513 …input, input_h, input_c, params, sequence_lengths, output, output_h, output_c, output_backprop, ou…
5746 … tf.Output, input_h tf.Output, input_c tf.Output, params tf.Output, sequence_lengths tf.Output, op…
5757 input, input_h, input_c, params, sequence_lengths,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 9932 name: "sequence_lengths"
10785 name: "sequence_lengths"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 17042 name: "sequence_lengths"
17630 name: "sequence_lengths"
|