Home
last modified time | relevance | path

Searched +path:pytorch +path:torch +path:distributed +path:algorithms (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/
H A D__init__.py
H A Djoin.py
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/model_averaging/
H A D__init__.py
H A Dutils.py
H A Dhierarchical_model_averager.py
H A Daveragers.py
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/
H A DpowerSGD_hook.py
H A Doptimizer_overlap_hooks.py
H A Dpost_localSGD_hook.py
H A Dmixed_precision_hooks.py
H A Dquantization_hooks.py
H A Ddefault_hooks.py
H A D__init__.py
H A Ddebugging_hooks.py
H A Dddp_zero_hook.py
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_quantization/
H A D__init__.py
H A Dquantization.py
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_optimizer_overlap/
H A Doptimizer_overlap.py
H A D__init__.py
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_checkpoint/
H A Dcheckpoint_wrapper.py
H A D__init__.py
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_comm_hooks/
H A D__init__.py
H A Ddefault_hooks.py