Searched refs:tf_gpu_dst (Results 1 – 1 of 1) sorted by relevance
1552 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities() local1555 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1559 ilink->set_device_id(tf_gpu_dst.value()); in GetDeviceLocalities()1568 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities() local1569 if (tf_gpu_id == tf_gpu_dst) continue; in GetDeviceLocalities()1572 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1575 ilink->set_device_id(tf_gpu_dst.value()); in GetDeviceLocalities()