| /external/tensorflow/tensorflow/core/ops/ |
| cudnn_rnn_ops_test.cc | 106 int max_seq_length = 2; local
111 std::vector<int> input_shape = {max_seq_length, batch_size, num_units};
114 std::vector<int> output_shape = {max_seq_length, batch_size,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_rnn_ops.cc | 151 int max_seq_length, int batch_size, int dir_count,
157 seq_length_(max_seq_length),
166 HashList({num_layers, input_size, num_units, max_seq_length, batch_size,
503 int max_seq_length; member in struct:tensorflow::TFRNNInputMode::__anon45229::CudnnRnnModelShapes
515 "[num_layers, input_size, num_units, dir_count, max_seq_length, "
517 num_layers, input_size, num_units, dir_count, max_seq_length,
577 model_shapes->max_seq_length = (*input)->dim_size(0);
580 model_shapes->max_seq_length = (*input)->dim_size(1);
627 TensorShape({model_shapes->max_seq_length, model_shapes->batch_size,
631 TensorShape({model_shapes->batch_size, model_shapes->max_seq_length,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1342 int max_seq_length() const { return max_seq_length_; } function in class:stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1404 int max_seq_length = 0; member in struct:stream_executor::gpu::__anon46392::RnnModelDims
[all...] |
