Searched refs:_scale_grad (Results 1 – 1 of 1) sorted by relevance
146 None if g is None else self._scale_grad(g, loss_scale_reciprocal)150 def _scale_grad(self, grad, loss_scale_reciprocal): member in MixedPrecisionLossScaleOptimizer