Searched refs:platform_gpu_dst (Results 1 – 1 of 1) sorted by relevance
1553 PlatformGpuId platform_gpu_dst; in GetDeviceLocalities() local1555 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1556 if (imap.directed_links.find({platform_gpu_id, platform_gpu_dst}) != in GetDeviceLocalities()1570 PlatformGpuId platform_gpu_dst; in GetDeviceLocalities() local1572 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()1573 if (platform_gpu_id == platform_gpu_dst) { in GetDeviceLocalities()