Searched refs:num_gpus_ (Results 1 – 2 of 2) sorted by relevance
77 num_gpus_ = GetNumAvailableGPUs(); in Initialize()78 LOG(INFO) << "Number of GPUs: " << num_gpus_; in Initialize()225 if (num_gpus_ > 0) { in AddOneReplica()226 new_node->set_device(strings::StrCat("/gpu:", number % num_gpus_)); in AddOneReplica()
50 int num_gpus_; variable