Searched refs:_compute_power (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 278 def _compute_power(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name=None): member in ShampooOptimizer
369 mat_h = self._compute_power(var, mat_g_updated, shape[i], neg_alpha)
373 lambda: self._compute_power(var, mat_g_updated, shape[i],
|