Home
last modified time | relevance | path

Searched defs:CudnnRnnModelShapes (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc498 struct CudnnRnnModelShapes { struct
499 int num_layers;
500 int input_size;
501 int num_units;
502 int dir_count;
503 int max_seq_length;
504 int batch_size;
505 TensorShape input_shape;
506 TensorShape output_shape;
507 TensorShape hidden_state_shape;
[all …]