Searched refs:MixedPrecisionLossScaleOptimizer (Results 1 – 10 of 10) sorted by relevance
137 opt = loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(157 opt = loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(opt, loss_scale)178 opt = loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(210 opt = loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(246 opt = loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(285 opt = loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(314 loss_scale_optimizer.MixedPrecisionLossScaleOptimizer(opt, None)
76 opt, loss_scale_optimizer_v1.MixedPrecisionLossScaleOptimizer)94 opt = loss_scale_optimizer_v1.MixedPrecisionLossScaleOptimizer(opt,
35 class MixedPrecisionLossScaleOptimizer(optimizer.Optimizer): class72 super(MixedPrecisionLossScaleOptimizer, self).__init__(use_locking, name)
34 loss_scale_optimizer_v1.MixedPrecisionLossScaleOptimizer,
1 path: "tensorflow.train.experimental.MixedPrecisionLossScaleOptimizer"3 …\'tensorflow.python.training.experimental.loss_scale_optimizer.MixedPrecisionLossScaleOptimizer\'>"
1 path: "tensorflow.mixed_precision.MixedPrecisionLossScaleOptimizer"3 …\'tensorflow.python.training.experimental.loss_scale_optimizer.MixedPrecisionLossScaleOptimizer\'>"
16 name: "MixedPrecisionLossScaleOptimizer"
35 …ensorflow.python.training.experimental.loss_scale_optimizer import MixedPrecisionLossScaleOptimizer240 optimizer = MixedPrecisionLossScaleOptimizer(AdamOptimizer(), 'dynamic')
36 …ensorflow.python.training.experimental.loss_scale_optimizer import MixedPrecisionLossScaleOptimizer