Home
last modified time | relevance | path

Searched +path:distributed +path:algorithms (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/
H A D__init__.py
H A Djoin.py
/aosp_15_r20/external/pytorch/test/distributed/algorithms/
H A Dtest_join.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
/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/test/distributed/algorithms/quantization/
H A Dtest_quantization.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
/aosp_15_r20/external/pytorch/test/distributed/algorithms/ddp_comm_hooks/
H A Dtest_ddp_hooks.py

12