Searched refs: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
256 if mat_h_slot_name is not None:
257 return state_ops.assign(self.get_slot(var, mat_h_slot_name), mat_h)
260 def _compute_power_iter(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name, argument
274 if mat_h_slot_name is not None:
275 return state_ops.assign(self.get_slot(var, mat_h_slot_name), mat_h)
278 def _compute_power(self, var, mat_g, mat_g_size, alpha, mat_h_slot_name=None): argument
283 mat_h_slot_name)
286 mat_h_slot_name)
|