Searched refs:_get_scale_async (Results 1 – 1 of 1) sorted by relevance
425 scaler = self._get_scale_async()533 def _get_scale_async(self) -> Optional[torch.Tensor]: member in GradScaler545 if (scale := self._get_scale_async()) is None