Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/
H A Ddistributed_c10d.py257 backend_capability: Dict[str, List[str]] = { variable in Backend
340 Backend.backend_capability[name.lower()] = ["cpu", "cuda"]
343 Backend.backend_capability[name.lower()] = [devices]
345 Backend.backend_capability[name.lower()] = devices
372 supported_devices = Backend.backend_capability[backend.lower()]