Home
last modified time | relevance | path

Searched refs:_power_method (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/nn/utils/
Dparametrizations.py437 self._power_method(weight_mat, 15)
452 def _power_method(self, weight_mat: torch.Tensor, n_power_iterations: int) -> None: member in _SpectralNorm
511 self._power_method(weight_mat, self.n_power_iterations)