Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1091 PlatformGpuId gpu_id_i = visible_gpu_order[i]; in CreateDevices() local
1094 if (im.directed_links.find({gpu_id_i, gpu_id_j}) != in CreateDevices()
1264 for (PlatformGpuId gpu_id_i : visible_gpu_order) { in GetInterconnectMaps() local
1266 if (gpu_id_i == gpu_id_j) continue; in GetInterconnectMaps()
1267 if ((*access_map)[{gpu_id_i, gpu_id_j}]) { in GetInterconnectMaps()
1268 imap.directed_links.insert({gpu_id_i, gpu_id_j}); in GetInterconnectMaps()