Home
last modified time | relevance | path

Searched refs:tf_gpu_dst (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1314 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities() local
1317 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() local
1331 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()