| /external/tensorflow/tensorflow/core/ops/ |
| cudnn_rnn_ops_test.cc | 106 int max_seq_length = 2; local
111 std::vector<int> input_shape = {max_seq_length, batch_size, num_units};
114 std::vector<int> output_shape = {max_seq_length, batch_size,
|
| cudnn_rnn_ops.cc | 174 auto max_seq_length = c->Dim(input_shape, 0);
185 c->MakeShape({max_seq_length, batch_size, output_size});
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1342 int max_seq_length() const { return max_seq_length_; } function in class:stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1404 int max_seq_length = 0; member in struct:stream_executor::gpu::__anon46392::RnnModelDims
[all...] |
| cuda_dnn.h | 58 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
63 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| stream_executor_pimpl.cc | 399 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length,
408 max_seq_length, batch_size, data_size, data_type);
413 int max_seq_length, int batch_size, int data_size,
422 max_seq_length, batch_size, data_size, seq_lengths, time_major,
|
| stream_executor_pimpl.h | 417 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
421 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
[all...] |
| dnn.h | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_rnn_ops.cc | 151 int max_seq_length, int batch_size, int dir_count,
157 seq_length_(max_seq_length),
166 HashList({num_layers, input_size, num_units, max_seq_length, batch_size,
503 int max_seq_length; member in struct:tensorflow::TFRNNInputMode::__anon45229::CudnnRnnModelShapes
515 "[num_layers, input_size, num_units, dir_count, max_seq_length, "
517 num_layers, input_size, num_units, dir_count, max_seq_length,
577 model_shapes->max_seq_length = (*input)->dim_size(0);
580 model_shapes->max_seq_length = (*input)->dim_size(1);
627 TensorShape({model_shapes->max_seq_length, model_shapes->batch_size,
631 TensorShape({model_shapes->batch_size, model_shapes->max_seq_length,
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| ctc_ops.py | 363 label_seq: tensor of shape [batch_size, max_seq_length]
415 def ctc_state_log_probs(seq_lengths, max_seq_length):
423 max_seq_length: int, max sequence length possible.
430 num_label_states = max_seq_length + 1
797 tuple of Tensor of shape (batch, max_seq_length) with repeated labels
798 collapsed and padded to max_seq_length, eg:
[all...] |
