Searched defs:mat_g_size (Results 1 – 2 of 2) 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
|
| D | matrix_functions.py | 77 mat_g_size, argument
|