Searched defs:flatten_parameters (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/benchmarks/functional_autograd_benchmark/ | ||
| D | torchaudio_models.py | 204 def flatten_parameters(self): member in BatchRNN |
| /external/pytorch/torch/nn/modules/ | ||
| D | rnn.py | 224 def flatten_parameters(self) -> None: member in RNNBase |
| /external/pytorch/torch/csrc/api/src/nn/modules/ | ||
| D | rnn.cpp | 180 void RNNImplBase<Derived>::flatten_parameters() { in flatten_parameters() function in torch::nn::detail::RNNImplBase |