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 __anon3a5366b80502() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 384 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
398 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| D | dnn.h | 2132 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
2139 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 510 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1456 CudnnRnnSequenceTensorDescriptor(GpuExecutor* parent, int max_seq_length, in CudnnRnnSequenceTensorDescriptor()
1475 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1494 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1533 int max_seq_length() const { return max_seq_length_; } in max_seq_length() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1589 int max_seq_length = 0; member
2093 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
2106 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()
500 int max_seq_length; member
|