Searched refs:mat_gbar_decay (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 76 mat_gbar_decay=1.0, argument
128 self._mat_gbar_decay = mat_gbar_decay
203 def _update_mat_g(self, mat_g, grad, axes, mat_gbar_decay, argument
228 return self._weighted_average(mat_g, self._mat_gbar_decay, mat_gbar_decay,
|
| D | weight_decay_optimizers.py | 398 mat_gbar_decay=1.0, argument
444 mat_gbar_decay=mat_gbar_weight,
|