Searched refs:NumGPUs (Results 1 – 1 of 1) sorted by relevance
84 static int32 NumGPUs() { return static_cast<int32>(devices_->size()); } in NumGPUs() function in tensorflow::NcclManagerTest89 const auto num_gpus = NumGPUs(); in PopulateMultiNodeParams()468 const int num_ranks = this->NumGPUs(); in TYPED_TEST()503 const int num_ranks = this->NumGPUs(); in TYPED_TEST()578 const int num_ranks = this->NumGPUs(); in TYPED_TEST()608 /*num_ranks_per_node=*/this->NumGPUs(), in TYPED_TEST()616 /*num_ranks_per_node=*/this->NumGPUs(), in TYPED_TEST()623 for (int num_ranks = 1; num_ranks <= this->NumGPUs(); ++num_ranks) { in TYPED_TEST()663 /*num_ranks_per_node=*/this->NumGPUs()); in TYPED_TEST()