Searched refs:cuda_device_id (Results 1 – 1 of 1) sorted by relevance
436 int cuda_device_id = 0; in CreateTRTNode() local439 cuda_device_id = device_allocator.first; in CreateTRTNode()446 cudaSetDevice(cuda_device_id); in CreateTRTNode()633 int cuda_device_id = -1; in GetDeviceAndAllocator() local641 cuda_device_id = platform_gpu_id.value(); in GetDeviceAndAllocator()642 if (cuda_device_id >= 0) { in GetDeviceAndAllocator()650 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()670 cuda_device_id = devices[0]->tensorflow_gpu_device_info()->gpu_id; in GetDeviceAndAllocator()673 << " and cuda_device_id " << cuda_device_id; in GetDeviceAndAllocator()678 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()[all …]