Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale.py434 def update_if_not_finite_grads(): function
444 update_if_not_finite_grads)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py375 def update_if_not_finite_grads(): function
385 update_if_not_finite_grads)