Home
last modified time | relevance | path

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

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