Searched defs:get_world_size (Results 1 – 4 of 4) sorted by relevance
82 def get_world_size(self) -> int: member in _DistWrapper
637 def get_world_size(): function
498 int CUDASymmetricMemory::get_world_size() { in get_world_size() function in c10d::symmetric_memory::CUDASymmetricMemory
2004 def get_world_size(group: Optional[ProcessGroup] = None) -> int: function