Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale.py306 def _assign_if_finite(var, value): function
426 _assign_if_finite(self._current_loss_scale, new_loss_scale),
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py208 def _assign_if_finite(var, value): function
367 _assign_if_finite(self.current_loss_scale, new_loss_scale),