Searched defs:CudnnRnnModelShapes (Results 1 – 1 of 1) sorted by relevance
497 struct CudnnRnnModelShapes { struct498 int num_layers;499 int input_size;500 int num_units;501 int dir_count;502 int max_seq_length;503 int batch_size;504 int cell_num_units = 0;508 TensorShape input_shape;509 TensorShape output_shape;[all …]