Searched defs:max_seq_length (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops_test.cc | 116 int max_seq_length = 2; in TEST() local
158 int max_seq_length = 2; in TEST() local
|
| D | cudnn_rnn_ops.cc | 194 auto max_seq_length = c->Dim(input_shape, 0); in __anon20e2ac3d0502() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 394 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
408 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| D | dnn.h | 2060 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
2067 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 556 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1635 CudnnRnnSequenceTensorDescriptor(GpuExecutor* parent, int max_seq_length, in CudnnRnnSequenceTensorDescriptor()
1653 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1671 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1710 int max_seq_length() const { return max_seq_length_; } in max_seq_length() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1764 int max_seq_length = 0; member
2436 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
2449 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 154 int max_seq_length, int batch_size, int dir_count, in CudnnRnnParameters()
503 int max_seq_length; member
|