Searched refs:num_tf_gpus (Results 1 – 2 of 2) sorted by relevance
1139 const int num_tf_gpus = next_tf_gpu_id; in CreateDevices() local1143 GetDeviceLocalities(num_tf_gpus, interconnect_maps, &device_localities)); in CreateDevices()1147 for (int di = 0; di < num_tf_gpus; ++di) { in CreateDevices()1276 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects, in GetDeviceLocalities() argument1279 all_tf_gpu_ids.reserve(num_tf_gpus); in GetDeviceLocalities()1280 for (int i = 0; i < num_tf_gpus; ++i) { in GetDeviceLocalities()
293 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects,