Searched defs:num_tf_gpus (Results 1 – 1 of 1) sorted by relevance
1357 const int num_tf_gpus = next_tf_gpu_id; in CreateDevices() local1403 int64 memory_limit, const DeviceLocality& dev_locality, size_t num_tf_gpus, in CreateGPUDevice()1510 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects, in GetDeviceLocalities()