Home
last modified time | relevance | path

Searched defs:_refresh_per_optimizer_state (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dsharded_grad_scaler.py15 def _refresh_per_optimizer_state() -> Dict[str, Any]: function
/aosp_15_r20/external/pytorch/torch/amp/
H A Dgrad_scaler.py45 def _refresh_per_optimizer_state() -> Dict[str, Any]: function