Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1586 struct RnnModelDims { struct
1587 int num_layers = 0;
1588 int batch_size = 0;
1589 int max_seq_length = 0;
1590 int hidden_size = 0;
1591 int input_size = 0;
1592 int cell_size = 0;
1593 int dir_count = 0;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1962 struct RnnModelDims { struct
1963 int num_layers = 0;
1964 int batch_size = 0;
1965 int seq_length = 0;
1966 int hidden_size = 0;
1967 int input_size = 0;
1968 int dir_count = 0;