Searched refs:current_loss_scale (Results 1 – 1 of 1) sorted by relevance
315 def current_loss_scale(self): member in _DynamicLossScaleState365 new_loss_scale = self.current_loss_scale * self.multiplier367 _assign_if_finite(self.current_loss_scale, new_loss_scale),379 self.current_loss_scale / self.multiplier, 1)382 self.current_loss_scale.assign(new_loss_scale))570 return ops.convert_to_tensor(self._loss_scale.current_loss_scale)