Searched defs:num_tf_gpus (Results 1 – 1 of 1) sorted by relevance
1408 const int num_tf_gpus = next_tf_device_id; in CreateDevices() local1450 const DeviceLocality& dev_locality, size_t num_tf_gpus, in CreateGPUDevice()1559 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects, in GetDeviceLocalities()