Searched refs:_scale_loss (Results 1 – 2 of 2) sorted by relevance
118 loss = self._scale_loss(loss)132 def _scale_loss(self, loss): member in MixedPrecisionLossScaleOptimizer
476 loss_value = self._scale_loss(loss_value)493 loss = self._scale_loss(loss)530 def _scale_loss(loss_value): member in Optimizer