Searched defs:mat_h_slot_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 231 def _compute_power_svd(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name): argument
260 def _compute_power_iter(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name, argument
278 def _compute_power(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name=None): argument
|