Searched defs:max_seq_length (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops_test.cc | 106 int max_seq_length = 2; in TEST() local
|
| D | cudnn_rnn_ops.cc | 174 auto max_seq_length = c->Dim(input_shape, 0); in __anone1d60d010502() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 415 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 399 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
413 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1253 CudnnRnnSequenceTensorDescriptor(GpuExecutor* parent, int max_seq_length, in CudnnRnnSequenceTensorDescriptor()
1276 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1295 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1342 int max_seq_length() const { return max_seq_length_; } in max_seq_length() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1404 int max_seq_length = 0; member
1838 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
1851 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 151 int max_seq_length, int batch_size, int dir_count, in CudnnRnnParameters()
503 int max_seq_length; member
|