Searched defs:_reset_parameters (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/modules/ | ||
| H A D | transformer.py | 302 def _reset_parameters(self): member in Transformer |
| H A D | activation.py | 1111 def _reset_parameters(self): member in MultiheadAttention |
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ | ||
| H A D | activation.cpp | 543 void MultiheadAttentionImpl::_reset_parameters() { in _reset_parameters() function in torch::nn::MultiheadAttentionImpl |