Searched refs:num_gpus_ (Results 1 – 5 of 5) sorted by relevance
108 : num_gpus_(NumGpus()), cupti_interface_(cupti_interface) {} in CuptiTracer()120 int num_gpus_; variable
637 num_gpus_(option.num_gpus), in CuptiTraceCollectorImpl()641 if (event.device_id >= num_gpus_) return; in AddEvent()666 for (int i = 0; i < num_gpus_; ++i) { in Export()679 for (int device_ordinal = 0; device_ordinal < num_gpus_; ++device_ordinal) { in Export()726 int num_gpus_; member in tensorflow::profiler::CuptiTraceCollectorImpl
1807 if (device_id >= num_gpus_) { in HandleCallback()1822 for (int i = 0; i < num_gpus_; ++i) { in HandleCallback()
78 num_gpus_ = GetNumAvailableGPUs(); in Initialize()79 LOG(INFO) << "Number of GPUs: " << num_gpus_; in Initialize()228 if (num_gpus_ > 0) { in AddOneReplica()229 new_node->set_device(strings::StrCat("/gpu:", number % num_gpus_)); in AddOneReplica()
52 int num_gpus_; variable