Home
last modified time | relevance | path

Searched defs:flatten_parameters (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/
H A Dtorchaudio_models.py204 def flatten_parameters(self): member in BatchRNN
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp180 void RNNImplBase<Derived>::flatten_parameters() { in flatten_parameters() function in torch::nn::detail::RNNImplBase
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Drnn.py224 def flatten_parameters(self) -> None: member in RNNBase