Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py5584 def _test_device_maps_multi_gpu(self, dst): member in TensorPipeAgentCudaRpcTest
5614 self._test_device_maps_multi_gpu(dst)
5619 self._test_device_maps_multi_gpu(dst)