Searched refs:gpu_manager (Results 1 – 3 of 3) sorted by relevance
44 se::Platform* gpu_manager = GPUMachineManager(); in GetNumAvailableGPUs() local45 if (gpu_manager != nullptr) { in GetNumAvailableGPUs()46 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs()49 auto desc = gpu_manager->DescriptionForDevice(i); in GetNumAvailableGPUs()
1213 se::Platform* gpu_manager = GPUMachineManager(); in CacheDeviceIds() local1214 if (gpu_manager == nullptr) { in CacheDeviceIds()1218 int device_count = gpu_manager->VisibleDeviceCount(); in CacheDeviceIds()1250 se::Platform* gpu_manager = GPUMachineManager(); in GetDeviceDetails() local1251 if (gpu_manager == nullptr) { in GetDeviceDetails()1255 gpu_manager->DescriptionForDevice(platform_device_id.value()); in GetDeviceDetails()1272 se::Platform* gpu_manager = GPUMachineManager(); in CreateDevices() local1273 if (gpu_manager == nullptr) { in CreateDevices()1277 if (gpu_manager->VisibleDeviceCount() <= 0) { in CreateDevices()1295 gpu_options.visible_device_list(), gpu_manager->VisibleDeviceCount(), in CreateDevices()[all …]
370 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps);