Searched refs:dynamic_growth_steps (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer_test.py | 135 self.assertIsNone(opt.dynamic_growth_steps)
184 dynamic_growth_steps=1)
187 self.assertEqual(opt.dynamic_growth_steps, 1)
188 self.assertIsInstance(opt.dynamic_growth_steps, int)
213 self.assertEqual(opt.dynamic_growth_steps, 2000)
227 dynamic_growth_steps=1)
267 dynamic_growth_steps=1)
298 dynamic_growth_steps=1)
323 opt, initial_scale=initial_scale, dynamic_growth_steps=1)
384 dynamic_growth_steps=1)
[all …]
|
| D | loss_scale_optimizer.py | 523 dynamic_growth_steps=None): argument
543 if dynamic_growth_steps is None:
544 dynamic_growth_steps = _DEFAULT_GROWTH_STEPS
546 initial_scale, dynamic_growth_steps, multiplier=2)
553 if dynamic_growth_steps is not None:
555 'is False, but got: %s' % (dynamic_growth_steps,))
604 def dynamic_growth_steps(self): member in LossScaleOptimizer
763 'dynamic_growth_steps': self.dynamic_growth_steps,
|
| D | keras_test.py | 838 opt, initial_scale=initial_loss_scale, dynamic_growth_steps=2)
1058 opt, initial_scale=1., dynamic_growth_steps=2.)
1184 dynamic_growth_steps=2.)
1222 self.assertEqual(model.optimizer.dynamic_growth_steps, 2.)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.mixed_precision.-loss-scale-optimizer.pbtxt | 25 name: "dynamic_growth_steps"
54 …[\'self\', \'inner_optimizer\', \'dynamic\', \'initial_scale\', \'dynamic_growth_steps\'], varargs…
|
| D | tensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt | 26 name: "dynamic_growth_steps"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.mixed_precision.-loss-scale-optimizer.pbtxt | 25 name: "dynamic_growth_steps"
54 …[\'self\', \'inner_optimizer\', \'dynamic\', \'initial_scale\', \'dynamic_growth_steps\'], varargs…
|
| D | tensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt | 26 name: "dynamic_growth_steps"
|