Home
last modified time | relevance | path

Searched refs:increment_period (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py128 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 …]
Dloss_scale.py354 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,
Dloss_scale_optimizer_test.py177 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/
Dloss_scale_benchmark.py78 increment_period = 1000000
80 initial_loss_scale=2., increment_period=increment_period)
Dloss_scale_optimizer.py781 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)
Dloss_scale_optimizer_test.py560 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.)
Dkeras_test.py827 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/
Dtensorflow.mixed_precision.-dynamic-loss-scale.pbtxt8 name: "increment_period"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
Dtensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt8 name: "increment_period"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
Dtensorflow.train.experimental.-dynamic-loss-scale.pbtxt8 name: "increment_period"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt8 name: "increment_period"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…
Dtensorflow.train.experimental.-dynamic-loss-scale.pbtxt8 name: "increment_period"
21 …argspec: "args=[\'self\', \'initial_loss_scale\', \'increment_period\', \'multiplier\'], varargs=N…