Searched refs:cuda_device_id (Results 1 – 1 of 1) sorted by relevance
732 int cuda_device_id = -1; in GetDeviceAndAllocator() local744 cuda_device_id = platform_gpu_id.value(); in GetDeviceAndAllocator()756 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()776 cuda_device_id = devices[0]->tensorflow_gpu_device_info()->gpu_id; in GetDeviceAndAllocator()779 << " and cuda_device_id " << cuda_device_id; in GetDeviceAndAllocator()784 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()910 int cuda_device_id = 0; in ConvertAfterShapes() local912 cuda_device_id = device_alloc.first; in ConvertAfterShapes()918 cudaSetDevice(cuda_device_id); in ConvertAfterShapes()