Searched defs:max_seq_length (Results 1 – 7 of 7) 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 | 177 auto max_seq_length = c->Dim(input_shape, 0); in __anon92a335e40502() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 391 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
405 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| D | dnn.h | 2034 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
2041 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 544 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1562 CudnnRnnSequenceTensorDescriptor(GpuExecutor* parent, int max_seq_length, in CudnnRnnSequenceTensorDescriptor()
1580 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1598 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1637 int max_seq_length() const { return max_seq_length_; } in max_seq_length() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1706 int max_seq_length = 0; member
2376 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
2389 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 153 int max_seq_length, int batch_size, int dir_count, in CudnnRnnParameters()
502 int max_seq_length; member
|