Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_pg_wrapper.py242 pg = c10d._create_process_group_wrapper(
404 pg = c10d._create_process_group_wrapper(
/aosp_15_r20/external/pytorch/torch/distributed/
H A D__init__.py132 _create_process_group_wrapper,
H A Ddistributed_c10d.py1840 backend_class = _create_process_group_wrapper(
4240 def _create_process_group_wrapper( function