Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2016 struct RnnModelDims { struct
2017 int num_layers = 0;
2018 int batch_size = 0;
2019 int seq_length = 0;
2020 int hidden_size = 0;
2021 int input_size = 0;
2022 int dir_count = 0;
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1761 struct RnnModelDims { struct
1762 int num_layers = 0;
1763 int batch_size = 0;
1764 int max_seq_length = 0;
1765 int hidden_size = 0;
1766 int input_size = 0;
1767 int cell_size = 0;
1768 int dir_count = 0;