Searched refs:_weighted_average (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 178 def _weighted_average(self, var, weight, weight_t, rest): member in ShampooOptimizer
228 return self._weighted_average(mat_g, self._mat_gbar_decay, mat_gbar_decay,
331 gbar_updated = self._weighted_average(gbar, self._gbar_decay,
406 mat_g_updated = self._weighted_average(mat_g,
|