Type LossScaleManager
Namespace tensorflow.contrib.mixed_precision
Public instance methods
object get_loss_scale()
Returns the loss scale.
object get_loss_scale_dyn()
Returns the loss scale.
object update_loss_scale(IEnumerable<object> finite_grads)
Updates loss scale based on if gradients are finite in current step.
object update_loss_scale_dyn(object finite_grads)
Updates loss scale based on if gradients are finite in current step.