Searched defs:mat_g (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 203 def _update_mat_g(self, mat_g, grad, axes, mat_gbar_decay, argument
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
|
| D | matrix_functions_test.py | 29 def np_power(mat_g, alpha): argument
|
| D | matrix_functions.py | 76 def matrix_inverse_pth_root(mat_g, argument
|
| D | shampoo_test.py | 37 def np_power(mat_g, alpha): argument
|