Searched defs:dropout_state (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/backends/cudnn/ |
| H A D | rnn.py | 45 def init_dropout_state(dropout, train, dropout_seed, dropout_state): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/ |
| H A D | AutocastRNN.cpp | 37 const std::optional<Tensor>& dropout_state) { in _cudnn_rnn_cast_reflatten()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 772 const Tensor& dropout_state = c10::value_or_else(dropout_state_opt, [] {return Tensor();}); in miopen_rnn_backward() local
832 Tensor dropout_state = at::empty({0}, input.options()); in _miopen_impl() local
851 Tensor dropout_state = at::empty({0}, input.options()); in _miopen_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 127 Tensor dropout_state; member
2121 const Tensor& dropout_state = in _cudnn_rnn_backward() local
2537 auto& dropout_state = get_dropout_state(dropout_p, train, input.options()); in _cudnn_impl() local
2601 auto& dropout_state = get_dropout_state(dropout_p, train, input.options()); in _cudnn_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 72 bool use_miopen(const at::Tensor& input, const double dropout_state) { in use_miopen()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 5933 dropout_state, argument
|