Home
last modified time | relevance | path

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

/external/pytorch/torch/nn/utils/
Dspectral_norm.py51 def reshape_weight_to_matrix(self, weight: torch.Tensor) -> torch.Tensor: member in SpectralNorm
94 weight_mat = self.reshape_weight_to_matrix(weight)
163 weight_mat = fn.reshape_weight_to_matrix(weight)
240 weight_mat = fn.reshape_weight_to_matrix(weight_orig)