Home
last modified time | relevance | path

Searched refs:precond_update_interval (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/opt/python/training/
Dshampoo.py80 precond_update_interval=1, argument
132 self._precond_update_interval = precond_update_interval
347 precond_update_interval = GetParam(self._precond_update_interval,
360 math_ops.mod(global_step, precond_update_interval) < 1,
363 mat_gbar_weight_t * precond_update_interval, i),
Dweight_decay_optimizers.py402 precond_update_interval=1, argument
447 precond_update_interval=precond_update_interval,
Dshampoo_test.py699 precond_update_interval = 5
716 precond_update_interval=precond_update_interval,
734 if (i + 1) % precond_update_interval == 0:
737 grad_np[i].shape[0] * precond_update_interval)
740 grad_np[i].shape[1] * precond_update_interval)
743 grad_np[i].shape[2] * precond_update_interval)