Searched defs:num_tf_gpus (Results 1 – 1 of 1) sorted by relevance
1562 const DeviceLocality& dev_locality, size_t num_tf_gpus, in CreateGPUDevice()1671 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects, in GetDeviceLocalities()