Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1139 const int num_tf_gpus = next_tf_gpu_id; in CreateDevices() local
1143 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() argument
1279 all_tf_gpu_ids.reserve(num_tf_gpus); in GetDeviceLocalities()
1280 for (int i = 0; i < num_tf_gpus; ++i) { in GetDeviceLocalities()
Dgpu_device.h293 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects,