Searched refs:original_device (Results 1 – 2 of 2) sorted by relevance
237 original_device = new_op.device247 with _MaybeDevice(original_device):
1014 int original_device = 0; in CreateDevices() local1016 cudaError_t err = cudaGetDevice(&original_device); in CreateDevices()1022 hipError_t err = hipGetDevice(&original_device); in CreateDevices()1062 err = cudaSetDevice(original_device); in CreateDevices()1064 return errors::Internal("cudaSetDevice() on GPU:", original_device, in CreateDevices()1068 err = hipSetDevice(original_device); in CreateDevices()1070 return errors::Internal("hipSetDevice() on GPU:", original_device, in CreateDevices()