Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/
H A Ddistributed_c10d.py423 def get_device_backend_map(self) -> Dict[str, Backend]: member in BackendConfig
1716 for device, backend_str in backend_config.get_device_backend_map().items():
1850 if len(set(backend_config.get_device_backend_map().values())) == 1:
1851 for device in backend_config.get_device_backend_map().keys():