Searched refs:inner_optimizer (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.mixed_precision.-loss-scale-optimizer.pbtxt | 37 name: "inner_optimizer"
54 …argspec: "args=[\'self\', \'inner_optimizer\', \'dynamic\', \'initial_scale\', \'dynamic_growth_st…
|
| D | tensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt | 38 name: "inner_optimizer"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.mixed_precision.-loss-scale-optimizer.pbtxt | 37 name: "inner_optimizer"
54 …argspec: "args=[\'self\', \'inner_optimizer\', \'dynamic\', \'initial_scale\', \'dynamic_growth_st…
|
| D | tensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt | 38 name: "inner_optimizer"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 522 def __init__(self, inner_optimizer, dynamic=True, initial_scale=None, argument
524 if not isinstance(inner_optimizer, optimizer_v2.OptimizerV2):
526 'but got: %s' % inner_optimizer)
534 self._optimizer = inner_optimizer
618 def inner_optimizer(self): member in LossScaleOptimizer
|
| D | loss_scale_optimizer_test.py | 459 self.assertIn('nesterov', dir(lso.inner_optimizer))
779 self.assertEqual(self.evaluate(opt.inner_optimizer.lr), 2.)
846 self.assertEqual(self.evaluate(opt.inner_optimizer.lr), 2.)
915 self.assertEqual(self.evaluate(opt.inner_optimizer.momentum), 0.5)
950 self.assertEqual(self.evaluate(opt.inner_optimizer.momentum), 0.5)
953 self.assertEqual(opt.inner_optimizer.my_attribute, 123)
|