Searched refs:platform_gpu_dst (Results 1 – 1 of 1) sorted by relevance
1315 PlatformGpuId platform_gpu_dst; in GetDeviceLocalities() local1317 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1318 if (imap.directed_links.find({platform_gpu_id, platform_gpu_dst}) != in GetDeviceLocalities()1332 PlatformGpuId platform_gpu_dst; in GetDeviceLocalities() local1334 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1335 if (platform_gpu_id == platform_gpu_dst) { in GetDeviceLocalities()