Searched refs:loss_scale_reciprocal (Results 1 – 2 of 2) sorted by relevance
144 loss_scale_reciprocal = 1 / loss_scale146 None if g is None else self._scale_grad(g, loss_scale_reciprocal)150 def _scale_grad(self, grad, loss_scale_reciprocal): argument152 grad_vals = grad.values * loss_scale_reciprocal154 return grad * loss_scale_reciprocal
671 loss_scale_reciprocal = 1. / self.loss_scale673 _multiply_gradient(g, loss_scale_reciprocal) if g is not None else None