Searched defs:CudnnRnnModelShapes (Results 1 – 1 of 1) sorted by relevance
498 struct CudnnRnnModelShapes { struct499 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 int cell_num_units = 0;509 TensorShape input_shape;510 TensorShape output_shape;[all …]