| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 177 initial_loss_scale=2, increment_period=1, multiplier=2)
181 loss_scale.initial_loss_scale % strategy.num_replicas_in_sync, 0)
209 initial_loss_scale=2, increment_period=1, multiplier=2)
241 initial_loss_scale = 2.
243 initial_loss_scale=initial_loss_scale,
258 self.assertEqual(self.evaluate(opt._loss_scale()), initial_loss_scale * 4)
268 self.evaluate(opt._loss_scale()), initial_loss_scale * 16)
283 initial_loss_scale=1., increment_period=2., multiplier=2.)
|
| D | loss_scale.py | 353 initial_loss_scale=2 ** 15, # See docstring for why this is big. argument
371 self._initial_loss_scale = float(initial_loss_scale)
385 def initial_loss_scale(self): member in DynamicLossScale
453 self.initial_loss_scale, self.increment_period, self.multiplier))
457 (self.initial_loss_scale, self.increment_period, self.multiplier))
461 'initial_loss_scale': self.initial_loss_scale,
|
| D | loss_scale_test.py | 127 initial_loss_scale=1., argument
131 initial_loss_scale=initial_loss_scale,
284 initial_loss_scale=1, increment_period=2, multiplier=3)
301 self.assertEqual(scalar.initial_loss_scale, scalar2.initial_loss_scale)
315 initial_loss_scale=1, increment_period=2, multiplier=3)
|
| D | loss_scaling_gradient_tape_test.py | 385 loss_scale = loss_scale_module.DynamicLossScale(initial_loss_scale=32)
406 loss_scale = loss_scale_module.DynamicLossScale(initial_loss_scale=32)
453 loss_scale = loss_scale_module.DynamicLossScale(initial_loss_scale=32)
473 loss_scale = loss_scale_module.DynamicLossScale(initial_loss_scale=32.0)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.mixed_precision.-dynamic-loss-scale.pbtxt | 12 name: "initial_loss_scale"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
| D | tensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt | 12 name: "initial_loss_scale"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
| D | tensorflow.train.experimental.-dynamic-loss-scale.pbtxt | 12 name: "initial_loss_scale"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt | 12 name: "initial_loss_scale"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
| D | tensorflow.train.experimental.-dynamic-loss-scale.pbtxt | 12 name: "initial_loss_scale"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 219 initial_loss_scale, argument
224 self._initial_loss_scale = float(initial_loss_scale)
303 def initial_loss_scale(self): member in _DynamicLossScaleState
599 return self._loss_scale.initial_loss_scale
780 config['initial_scale'] = loss_scale.initial_loss_scale
1047 if loss_scale.initial_loss_scale != _DEFAULT_INITIAL_SCALE:
1048 kwargs['initial_scale'] = loss_scale.initial_loss_scale
1050 loss_scale.initial_loss_scale)
|
| D | keras_test.py | 817 initial_loss_scale = 2.
819 expected_gradient = backend.variable([initial_loss_scale / batch_size],
827 initial_loss_scale=initial_loss_scale, increment_period=2)
831 initial_loss_scale=initial_loss_scale, increment_period=2)
838 opt, initial_scale=initial_loss_scale, dynamic_growth_steps=2)
1180 initial_loss_scale=1., increment_period=2.)
|
| D | loss_scale_benchmark.py | 80 initial_loss_scale=2., increment_period=increment_period)
|
| D | loss_scale_optimizer_test.py | 560 initial_loss_scale=4, increment_period=1, multiplier=2)
804 initial_loss_scale=2, increment_period=3)
903 initial_loss_scale=2., increment_period=3.)
|