Searched refs:tf_gpu_dst (Results 1 – 1 of 1) sorted by relevance
1314 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities() local1317 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1321 ilink->set_device_id(tf_gpu_dst.value()); in GetDeviceLocalities()1330 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities() local1331 if (tf_gpu_id == tf_gpu_dst) continue; in GetDeviceLocalities()1334 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1337 ilink->set_device_id(tf_gpu_dst.value()); in GetDeviceLocalities()