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.cc495 struct CudnnRnnModelShapes { struct
496 int num_layers;
497 int input_size;
498 int num_units;
499 int dir_count;
500 int max_seq_length;
501 int batch_size;
502 int cell_num_units = 0;
506 TensorShape input_shape;
507 TensorShape output_shape;
[all …]