Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_optimizer_overlap/
H A Doptimizer_overlap.py22 def decorator(target_overlapped_optim_cls): argument
23 if target_overlapped_optim_cls in _registered_overlapped_optims:
29 _registered_overlapped_optims[optim_cls] = target_overlapped_optim_cls
30 return target_overlapped_optim_cls