| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops_test.cc | 106 int max_seq_length = 2; in TEST() local
111 std::vector<int> input_shape = {max_seq_length, batch_size, num_units}; in TEST()
114 std::vector<int> output_shape = {max_seq_length, batch_size, in TEST()
|
| D | cudnn_rnn_ops.cc | 174 auto max_seq_length = c->Dim(input_shape, 0); in __anone1d60d010502() local
185 c->MakeShape({max_seq_length, batch_size, output_size}); in __anone1d60d010502()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1253 CudnnRnnSequenceTensorDescriptor(GpuExecutor* parent, int max_seq_length, in CudnnRnnSequenceTensorDescriptor() argument
1260 : max_seq_length_(max_seq_length), in CudnnRnnSequenceTensorDescriptor()
1268 handles_(max_seq_length, handle_.get()) { in CudnnRnnSequenceTensorDescriptor()
1276 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create() argument
1278 CHECK_GT(max_seq_length, 0); in Create()
1286 return CudnnRnnSequenceTensorDescriptor(parent, max_seq_length, batch_size, in Create()
1295 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create() argument
1299 CHECK_GT(max_seq_length, 0); in Create()
1319 /*maxSeqLength=*/max_seq_length, in Create()
1324 parent, max_seq_length, batch_size, data_size, data_type, in Create()
[all …]
|
| D | cuda_dnn.h | 58 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
63 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 151 int max_seq_length, int batch_size, int dir_count, in CudnnRnnParameters() argument
157 seq_length_(max_seq_length), in CudnnRnnParameters()
166 HashList({num_layers, input_size, num_units, max_seq_length, batch_size, in CudnnRnnParameters()
503 int max_seq_length; member
517 num_layers, input_size, num_units, dir_count, max_seq_length, in DebugString()
577 model_shapes->max_seq_length = (*input)->dim_size(0); in ExtractForwardInput()
580 model_shapes->max_seq_length = (*input)->dim_size(1); in ExtractForwardInput()
627 TensorShape({model_shapes->max_seq_length, model_shapes->batch_size, in ExtractForwardInput()
631 TensorShape({model_shapes->batch_size, model_shapes->max_seq_length, in ExtractForwardInput()
1470 model_shapes.num_units, model_shapes.max_seq_length, in MaybeAutoTune()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 399 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor() argument
408 max_seq_length, batch_size, data_size, data_type); in createRnnSequenceTensorDescriptor()
413 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor() argument
422 max_seq_length, batch_size, data_size, seq_lengths, time_major, in createRnnSequenceTensorDescriptor()
|
| D | stream_executor_pimpl.h | 417 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
421 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
|
| D | dnn.h | 2063 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
2070 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 415 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
430 num_label_states = max_seq_length + 1
|