Home
last modified time | relevance | path

Searched refs:LossScaleOptimizerV1 (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dmixed_precision_graph_rewrite_test.py68 opt, loss_scale_optimizer_v2.LossScaleOptimizerV1)
74 opt = loss_scale_optimizer_v2.LossScaleOptimizerV1(opt, 'dynamic')
Dloss_scale_optimizer_test.py498 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, loss_scale=2)
516 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, 'dynamic')
543 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, loss_scale)
562 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, loss_scale)
585 loss_scale_optimizer.LossScaleOptimizerV1(opt, loss_scale)
601 loss_scale_optimizer.LossScaleOptimizerV1(opt, MyLossScale())
742 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, 2)
768 opt = loss_scale_optimizer.LossScaleOptimizerV1.from_config(config)
805 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, loss_scale)
835 opt = loss_scale_optimizer.LossScaleOptimizerV1.from_config(config)
[all …]
Dloss_scale_optimizer.py940 class LossScaleOptimizerV1(LossScaleOptimizer): class
1028 super(LossScaleOptimizerV1, self).__init__(optimizer, dynamic=False,
1036 super(LossScaleOptimizerV1, self).__init__(optimizer, dynamic=False,
1043 super(LossScaleOptimizerV1, self).__init__(optimizer)
1066 super(LossScaleOptimizerV1, self).__init__(optimizer, **kwargs)
1155 LossScaleOptimizerV1)
Dkeras_test.py833 opt = loss_scale_optimizer.LossScaleOptimizerV1(
1181 opt = loss_scale_optimizer.LossScaleOptimizerV1(opt, loss_scale)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt3 …ce: "<class \'tensorflow.python.keras.mixed_precision.loss_scale_optimizer.LossScaleOptimizerV1\'>"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.mixed_precision.experimental.-loss-scale-optimizer.pbtxt3 …ce: "<class \'tensorflow.python.keras.mixed_precision.loss_scale_optimizer.LossScaleOptimizerV1\'>"
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py609 opt = lso.LossScaleOptimizerV1(opt, loss_scale)
Dtraining_v1.py1377 self.optimizer = loss_scale_optimizer.LossScaleOptimizerV1(