Searched refs:_update_mat_g (Results 1 – 1 of 1) sorted by relevance
203 def _update_mat_g(self, mat_g, grad, axes, mat_gbar_decay, member in ShampooOptimizer361 lambda: self._update_mat_g(