Searched defs:fn_mode (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 41 int64_t fn_mode, in _cudnn_rnn_flatten_weight()
57 int64_t fn_mode, in _cudnn_rnn()
1352 int64_t fn_mode, in _cudnn_rnn_flatten_weight()
1388 int64_t fn_mode, in _cudnn_rnn()
1678 int64_t fn_mode, in _cudnn_rnn_backward_input()
1908 int64_t fn_mode, 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()
561 int64_t fn_mode, int64_t fn_hidden_size, in miopen_rnn_backward_input()
676 int64_t fn_mode, int64_t fn_hidden_size, in miopen_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | AutocastRNN.cpp | 18 int64_t fn_mode, in miopen_rnn()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RNN.cpp | 198 static Tensor _shuffle_weight(const Tensor& weight, int64_t fn_mode) { in _shuffle_weight()
207 static Tensor _shuffle_bias(const Tensor& bias_ih, const Tensor& bias_hh, int64_t fn_mode) { in _shuffle_bias()
|