xref: /aosp_15_r20/external/pytorch/torch/amp/__init__.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1from .autocast_mode import (
2    _enter_autocast,
3    _exit_autocast,
4    autocast,
5    custom_bwd,
6    custom_fwd,
7    is_autocast_available,
8)
9from .grad_scaler import GradScaler
10