Home
last modified time | relevance | path

Searched refs:DynamicLossScale (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape_test.py84 loss_scale_module.DynamicLossScale],
104 loss_scale_module.DynamicLossScale],
124 loss_scale_module.DynamicLossScale],
156 loss_scale_module.DynamicLossScale],
177 loss_scale_module.DynamicLossScale],
222 loss_scale_module.DynamicLossScale],
248 loss_scale_module.DynamicLossScale],
260 loss_scale_module.DynamicLossScale],
273 loss_scale_module.DynamicLossScale],
304 loss_scale_module.DynamicLossScale],
[all …]
Dloss_scale_test.py130 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)
294 loss_scale = loss_scale_module.DynamicLossScale()
300 scalar2 = loss_scale_module.DynamicLossScale()
307 scalar = loss_scale_module.DynamicLossScale()
314 loss_scale = loss_scale_module.DynamicLossScale(
Dloss_scale_optimizer_test.py176 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.py324 class DynamicLossScale(LossScale): class
370 super(DynamicLossScale, self).__init__()
472 return DynamicLossScale()
Dloss_scale_optimizer.py82 return isinstance(self._loss_scale, loss_scale_module.DynamicLossScale)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale.py37 'DynamicLossScale': loss_scale_module.DynamicLossScale,
56 return loss_scale_module.DynamicLossScale()
Dpolicy_test.py129 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)
Dloss_scale_optimizer.py778 elif isinstance(loss_scale, loss_scale_module.DynamicLossScale):
1044 elif isinstance(loss_scale, loss_scale_module.DynamicLossScale):
1087 if (isinstance(config['loss_scale'], loss_scale_module.DynamicLossScale)
1099 config['loss_scale'] = loss_scale_module.DynamicLossScale(
Dloss_scale_benchmark.py79 loss_scale = loss_scale_module.DynamicLossScale(
Dloss_scale_optimizer_test.py559 loss_scale = tf_loss_scale_module.DynamicLossScale(
580 loss_scale = tf_loss_scale_module.DynamicLossScale(multiplier=4)
803 loss_scale = tf_loss_scale_module.DynamicLossScale(
902 loss_scale = tf_loss_scale_module.DynamicLossScale(
Dkeras_test.py826 loss_scale = loss_scale_module.DynamicLossScale(
830 loss_scale = loss_scale_module.DynamicLossScale(
1179 loss_scale = loss_scale_module.DynamicLossScale(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.mixed_precision.-dynamic-loss-scale.pbtxt1 path: "tensorflow.mixed_precision.DynamicLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.DynamicLossScale\'>"
Dtensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt1 path: "tensorflow.mixed_precision.experimental.DynamicLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.DynamicLossScale\'>"
Dtensorflow.train.experimental.-dynamic-loss-scale.pbtxt1 path: "tensorflow.train.experimental.DynamicLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.DynamicLossScale\'>"
Dtensorflow.mixed_precision.experimental.pbtxt4 name: "DynamicLossScale"
Dtensorflow.train.experimental.pbtxt4 name: "DynamicLossScale"
Dtensorflow.mixed_precision.pbtxt4 name: "DynamicLossScale"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt1 path: "tensorflow.mixed_precision.experimental.DynamicLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.DynamicLossScale\'>"
Dtensorflow.train.experimental.-dynamic-loss-scale.pbtxt1 path: "tensorflow.train.experimental.DynamicLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.DynamicLossScale\'>"
Dtensorflow.mixed_precision.experimental.pbtxt4 name: "DynamicLossScale"
Dtensorflow.train.experimental.pbtxt4 name: "DynamicLossScale"
/external/tensorflow/
DRELEASE.md424 …* When passing a `tf.mixed_precision.experimental.DynamicLossScale` to `tf.keras.mixed_precision.e…
425 the `DynamicLossScale`'s multiplier must be 2.
426 * When passing a `tf.mixed_precision.experimental.DynamicLossScale` to
429 …This means modifying the weights of the `DynamicLossScale` will no longer affect the weights of th…