Home
last modified time | relevance | path

Searched refs:update_if_finite_grads (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale.py420 def update_if_finite_grads(): function
443 update_op = control_flow_ops.cond(is_finite, update_if_finite_grads,
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py361 def update_if_finite_grads(): function
384 update_op = control_flow_ops.cond(is_finite, update_if_finite_grads,