Searched refs:reshape_weight_to_matrix (Results 1 – 1 of 1) sorted by relevance
51 def reshape_weight_to_matrix(self, weight: torch.Tensor) -> torch.Tensor: member in SpectralNorm94 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)