Searched refs:_test_device_maps_gpu (Results 1 – 1 of 1) sorted by relevance
5309 def _test_device_maps_gpu(self, x_from, y_from, z_to, device_map, dst=None, fn=None): member in TensorPipeAgentCudaRpcTest5341 self._test_device_maps_gpu(5351 self._test_device_maps_gpu(5361 self._test_device_maps_gpu(5371 self._test_device_maps_gpu(5381 self._test_device_maps_gpu(5391 self._test_device_maps_gpu(5400 self._test_device_maps_gpu(5409 self._test_device_maps_gpu(5418 self._test_device_maps_gpu([all …]