Searched defs:fn_num_layers (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 44 int64_t fn_num_layers, in _cudnn_rnn_flatten_weight()
60 int64_t fn_num_layers, in _cudnn_rnn()
1355 int64_t fn_num_layers, in _cudnn_rnn_flatten_weight()
1391 int64_t fn_num_layers, in _cudnn_rnn()
1681 int64_t fn_num_layers, in _cudnn_rnn_backward_input()
1911 int64_t fn_num_layers, in _cudnn_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 33 int64_t fn_mode, int64_t fn_hidden_size, int64_t fn_num_layers, in miopen_rnn()
448 int64_t fn_mode, int64_t fn_hidden_size, int64_t fn_num_layers, in miopen_rnn()
562 int64_t fn_num_layers, bool batch_first, double fn_dropout, in miopen_rnn_backward_input()
677 int64_t fn_num_layers, bool batch_first, double fn_dropout, in miopen_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | AutocastRNN.cpp | 20 int64_t fn_num_layers, in miopen_rnn()
|