Home
last modified time | relevance | path

Searched refs:_set_hyper (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dftrl.py120 self._set_hyper('learning_rate', learning_rate)
121 self._set_hyper('decay', self._initial_decay)
122 self._set_hyper('learning_rate_power', learning_rate_power)
123 self._set_hyper('l1_regularization_strength', l1_regularization_strength)
124 self._set_hyper('l2_regularization_strength', l2_regularization_strength)
125 self._set_hyper('beta', beta)
Dadam.py116 self._set_hyper('learning_rate', kwargs.get('lr', learning_rate))
117 self._set_hyper('decay', self._initial_decay)
118 self._set_hyper('beta_1', beta_1)
119 self._set_hyper('beta_2', beta_2)
368 self._set_hyper('learning_rate', kwargs.get('lr', learning_rate))
369 self._set_hyper('decay', self._initial_decay)
370 self._set_hyper('beta_1', beta_1)
371 self._set_hyper('beta_2', beta_2)
Dadamax.py100 self._set_hyper('learning_rate', kwargs.get('lr', learning_rate))
101 self._set_hyper('decay', self._initial_decay)
102 self._set_hyper('beta_1', beta_1)
103 self._set_hyper('beta_2', beta_2)
Dadadelta.py88 self._set_hyper('learning_rate', kwargs.get('lr', learning_rate))
89 self._set_hyper('decay', self._initial_decay)
90 self._set_hyper('rho', rho)
Dgradient_descent.py110 self._set_hyper("learning_rate", kwargs.get("lr", learning_rate))
111 self._set_hyper("decay", self._initial_decay)
118 self._set_hyper("momentum", momentum)
Dnadam.py83 self._set_hyper('learning_rate', kwargs.get('lr', learning_rate))
84 self._set_hyper('decay', self._initial_decay)
85 self._set_hyper('beta_1', beta_1)
86 self._set_hyper('beta_2', beta_2)
Drmsprop.py139 self._set_hyper("learning_rate", kwargs.get("lr", learning_rate))
140 self._set_hyper("decay", self._initial_decay)
141 self._set_hyper("rho", rho)
148 self._set_hyper("momentum", momentum)
Dadagrad.py74 self._set_hyper('learning_rate', kwargs.get('lr', learning_rate))
75 self._set_hyper('decay', self._initial_decay)
Doptimizer_v2.py774 def _set_hyper(self, name, value): member in OptimizerV2
853 self._set_hyper(name, value)
1476 setter=RestoredOptimizer._set_hyper # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py920 self._optimizer._set_hyper(name, value)
Dloss_scale_optimizer_test.py431 self._set_hyper('loss_scale', 123.)