Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dgradient_descent.py106 nesterov=False, argument
120 self.nesterov = nesterov
146 use_nesterov=self.nesterov)
184 use_nesterov=self.nesterov)
192 "nesterov": self.nesterov,
Dgradient_descent_test.py379 learning_rate=2.0, momentum=0.9, nesterov=True)
413 learning_rate=2.0, momentum=0.9, nesterov=True)
662 opt = gradient_descent.SGD(learning_rate=1.0, momentum=0.9, nesterov=True)
691 self.assertTrue(opt3.nesterov)
Doptimizer_v2_test.py817 opt_k_v1 = optimizer_v1.SGD(momentum=0.9, nesterov=True)
818 opt_k_v2 = gradient_descent.SGD(momentum=0.9, nesterov=True)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ApplyAdam.pbtxt80 If `True`, uses the nesterov update.
Dapi_def_ResourceApplyAdam.pbtxt74 If `True`, uses the nesterov update.
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py180 def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, **kwargs): argument
188 self.nesterov = nesterov
212 if self.nesterov:
229 'nesterov': self.nesterov
Doptimizers_test.py118 optimizer_v1.SGD(lr=0.01, momentum=0.9, nesterov=True))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.optimizers.-s-g-d.pbtxt29 …argspec: "args=[\'self\', \'learning_rate\', \'momentum\', \'nesterov\', \'name\'], varargs=None, …
Dtensorflow.optimizers.-s-g-d.pbtxt29 …argspec: "args=[\'self\', \'learning_rate\', \'momentum\', \'nesterov\', \'name\'], varargs=None, …
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.optimizers.-s-g-d.pbtxt29 …argspec: "args=[\'self\', \'learning_rate\', \'momentum\', \'nesterov\', \'name\'], varargs=None, …
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer_test.py441 self.assertFalse(opt.nesterov)
445 lso.nesterov # pylint: disable=pointless-statement
447 lso.nesterov = True
448 self.assertTrue(lso.nesterov)
449 self.assertFalse(opt.nesterov)