Searched refs:all_tf_gpu_ids (Results 1 – 1 of 1) sorted by relevance
1278 std::vector<TfGpuId> all_tf_gpu_ids; in GetDeviceLocalities() local1279 all_tf_gpu_ids.reserve(num_tf_gpus); in GetDeviceLocalities()1281 all_tf_gpu_ids.push_back(TfGpuId(i)); in GetDeviceLocalities()1283 for (TfGpuId tf_gpu_id : all_tf_gpu_ids) { in GetDeviceLocalities()1314 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities()1330 for (TfGpuId tf_gpu_dst : all_tf_gpu_ids) { in GetDeviceLocalities()