Searched refs:initial_scale (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer_test.py | 97 initial_scale=loss_scale)
119 initial_scale=loss_scale)
132 initial_scale=2.)
140 initial_scale=2.)
151 initial_scale=2)
163 initial_scale=2)
183 opt = loss_scale_optimizer.LossScaleOptimizer(opt, initial_scale=2,
185 self.assertEqual(opt.initial_scale, 2.)
186 self.assertIsInstance(opt.initial_scale, float)
190 self.assertEqual(opt.initial_scale % strategy.num_replicas_in_sync, 0)
[all …]
|
| D | loss_scale_optimizer.py | 522 def __init__(self, inner_optimizer, dynamic=True, initial_scale=None, argument
541 if initial_scale is None:
542 initial_scale = _DEFAULT_INITIAL_SCALE
546 initial_scale, dynamic_growth_steps, multiplier=2)
549 if initial_scale is None:
552 self._loss_scale = float(initial_scale)
592 def initial_scale(self): member in LossScaleOptimizer
762 'initial_scale': self.initial_scale,
1029 initial_scale=loss_scale)
1037 initial_scale=ls_val)
|
| D | keras_test.py | 590 initial_scale=8)
687 initial_scale=loss_scale)
766 opt, dynamic=False, initial_scale=loss_scale)
838 opt, initial_scale=initial_loss_scale, dynamic_growth_steps=2)
909 initial_scale=5.)
1022 initial_scale=1)
1058 opt, initial_scale=1., dynamic_growth_steps=2.)
1183 opt = loss_scale_optimizer.LossScaleOptimizer(opt, initial_scale=1.,
1221 self.assertEqual(model.optimizer.initial_scale, 1.)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.mixed_precision.-loss-scale-optimizer.pbtxt | 33 name: "initial_scale"
54 …argspec: "args=[\'self\', \'inner_optimizer\', \'dynamic\', \'initial_scale\', \'dynamic_growth_st…
|
| D | tensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt | 34 name: "initial_scale"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.mixed_precision.-loss-scale-optimizer.pbtxt | 33 name: "initial_scale"
54 …argspec: "args=[\'self\', \'inner_optimizer\', \'dynamic\', \'initial_scale\', \'dynamic_growth_st…
|
| D | tensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt | 34 name: "initial_scale"
|