Searched refs:increment_period (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/python/training/experimental/ |
D | loss_scale_test.py | 128 increment_period=2, argument 132 increment_period=increment_period, 216 increment_period = 1 222 increment_period) 230 increment_period = 1 235 increment_period) 243 increment_period = 2 248 increment_period) 284 initial_loss_scale=1, increment_period=2, multiplier=3) 289 self.assertEqual(loss_scale.increment_period, 2) [all …]
|
D | loss_scale.py | 354 increment_period=2000, argument 372 self._increment_period = int(increment_period) 389 def increment_period(self): member in DynamicLossScale 453 self.initial_loss_scale, self.increment_period, self.multiplier)) 457 (self.initial_loss_scale, self.increment_period, self.multiplier)) 462 'increment_period': self.increment_period,
|
D | loss_scale_optimizer_test.py | 177 initial_loss_scale=2, increment_period=1, multiplier=2) 209 initial_loss_scale=2, increment_period=1, multiplier=2) 244 increment_period=1, 283 initial_loss_scale=1., increment_period=2., multiplier=2.)
|
/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
D | loss_scale_benchmark.py | 78 increment_period = 1000000 80 initial_loss_scale=2., increment_period=increment_period)
|
D | loss_scale_optimizer.py | 781 config['dynamic_growth_steps'] = loss_scale.increment_period 1051 if loss_scale.increment_period != _DEFAULT_GROWTH_STEPS: 1052 kwargs['dynamic_growth_steps'] = loss_scale.increment_period 1054 loss_scale.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.)
|
D | keras_test.py | 827 initial_loss_scale=initial_loss_scale, increment_period=2) 831 initial_loss_scale=initial_loss_scale, increment_period=2) 1180 initial_loss_scale=1., increment_period=2.)
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.mixed_precision.-dynamic-loss-scale.pbtxt | 8 name: "increment_period" 21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
D | tensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt | 8 name: "increment_period" 21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
D | tensorflow.train.experimental.-dynamic-loss-scale.pbtxt | 8 name: "increment_period" 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 | 8 name: "increment_period" 21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|
D | tensorflow.train.experimental.-dynamic-loss-scale.pbtxt | 8 name: "increment_period" 21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
|