Searched refs:_compute_power_iter (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 260 def _compute_power_iter(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name, member in ShampooOptimizer
282 return self._compute_power_iter(var, mat_g, mat_g_size, alpha,
|