Home
last modified time | relevance | path

Searched defs:pg_group_ranks (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/testing/_internal/distributed/
Dmulti_threaded_pg.py496 def pg_group_ranks(self): member in ThreadLocalWorld
/external/pytorch/torch/distributed/
Ddistributed_c10d.py614 def pg_group_ranks(self) -> Dict[ProcessGroup, Dict[int, int]]: member in _World