_apply_optim_in_backward_hook (reference) in projects: aosp_15_r20

Searched refs:_apply_optim_in_backward_hook (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/
H A D	optimizer_overlap_hooks.py	`49 def _apply_optim_in_backward_hook( function`
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A D	distributed.py	`1036 _apply_optim_in_backward_hook, 1041 _apply_optim_in_backward_hook(`