Searched refs:_apply_gradient (Results 1 – 1 of 1) sorted by relevance
160 return self._apply_gradient(grad, var)175 return self._apply_gradient(dense_grad, var)176 return self._apply_gradient(grad_values, var, grad_indices)288 def _apply_gradient(self, grad, var, indices=None): member in ShampooOptimizer