Searched defs:fn_bidirectional (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 46 bool fn_bidirectional) { in _cudnn_rnn_flatten_weight()
64 bool fn_bidirectional, in _cudnn_rnn()
1357 bool fn_bidirectional) { in _cudnn_rnn_flatten_weight()
1395 bool fn_bidirectional, in _cudnn_rnn()
1685 bool fn_bidirectional, in _cudnn_rnn_backward_input()
1915 bool fn_bidirectional, in _cudnn_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 34 bool batch_first, double fn_dropout, bool fn_train, bool fn_bidirectional, in miopen_rnn()
449 bool batch_first, double fn_dropout, bool fn_train, bool fn_bidirectional, in miopen_rnn()
563 bool fn_train, bool fn_bidirectional, IntArrayRef fn_batch_sizes, in miopen_rnn_backward_input()
678 bool fn_train, bool fn_bidirectional, IntArrayRef fn_batch_sizes, in miopen_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | AutocastRNN.cpp | 24 bool fn_bidirectional, in miopen_rnn()
|