Home
last modified time | relevance | path

Searched refs:loss_scale_module (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape_test.py34 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
83 loss_scale=[loss_scale_module.FixedLossScale,
84 loss_scale_module.DynamicLossScale],
103 loss_scale=[loss_scale_module.FixedLossScale,
104 loss_scale_module.DynamicLossScale],
123 loss_scale=[loss_scale_module.FixedLossScale,
124 loss_scale_module.DynamicLossScale],
155 loss_scale=[loss_scale_module.FixedLossScale,
156 loss_scale_module.DynamicLossScale],
176 loss_scale=[loss_scale_module.FixedLossScale,
[all …]
Dloss_scale_test.py36 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
66 loss_scale = loss_scale_module.FixedLossScale(loss_scale_value)
85 loss_scale = loss_scale_module.get(123)
87 loss_scale = loss_scale_module.FixedLossScale.from_config(config)
92 scalar = loss_scale_module.FixedLossScale(123)
97 loss_scale = loss_scale_module.FixedLossScale(123)
130 loss_scale = loss_scale_module.DynamicLossScale(
276 loss_scale = loss_scale_module.DynamicLossScale()
283 loss_scale = loss_scale_module.DynamicLossScale(
286 loss_scale = loss_scale_module.DynamicLossScale.from_config(config)
[all …]
Dloss_scale_optimizer_test.py38 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
176 loss_scale = loss_scale_module.DynamicLossScale(
208 loss_scale = loss_scale_module.DynamicLossScale(
242 loss_scale = loss_scale_module.DynamicLossScale(
282 loss_scale = loss_scale_module.DynamicLossScale(
Dloss_scale_optimizer.py26 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
74 self._loss_scale = loss_scale_module.get(loss_scale)
82 return isinstance(self._loss_scale, loss_scale_module.DynamicLossScale)
Dloss_scaling_gradient_tape.py28 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
115 if not isinstance(loss_scale, loss_scale_module.LossScale):
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale.py27 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
36 'FixedLossScale': loss_scale_module.FixedLossScale,
37 'DynamicLossScale': loss_scale_module.DynamicLossScale,
54 return loss_scale_module.FixedLossScale(identifier)
56 return loss_scale_module.DynamicLossScale()
57 if isinstance(identifier, loss_scale_module.LossScale):
Dloss_scale_benchmark.py33 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
71 loss_scale = loss_scale_module.FixedLossScale(2.)
79 loss_scale = loss_scale_module.DynamicLossScale(
Dloss_scale_optimizer.py40 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
775 if isinstance(loss_scale, loss_scale_module.FixedLossScale):
778 elif isinstance(loss_scale, loss_scale_module.DynamicLossScale):
1030 elif isinstance(loss_scale, loss_scale_module.FixedLossScale):
1044 elif isinstance(loss_scale, loss_scale_module.DynamicLossScale):
1067 elif isinstance(loss_scale, loss_scale_module.LossScale):
1087 if (isinstance(config['loss_scale'], loss_scale_module.DynamicLossScale)
1099 config['loss_scale'] = loss_scale_module.DynamicLossScale(
1102 config['loss_scale'] = loss_scale_module.FixedLossScale(
Dpolicy_test.py35 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
129 ls = loss_scale_module.DynamicLossScale()
134 self.assertIsInstance(policy.loss_scale, loss_scale_module.DynamicLossScale)
137 self.assertIsInstance(policy.loss_scale, loss_scale_module.DynamicLossScale)
Dkeras_test.py59 from tensorflow.python.training.experimental import loss_scale as loss_scale_module unknown
826 loss_scale = loss_scale_module.DynamicLossScale(
830 loss_scale = loss_scale_module.DynamicLossScale(
1179 loss_scale = loss_scale_module.DynamicLossScale(