Searched refs:GetNumAvailableGPUs (Results 1 – 8 of 8) sorted by relevance
30 int GetNumAvailableGPUs();
30 int GetNumAvailableGPUs() { in GetNumAvailableGPUs() function
77 num_gpus_ = GetNumAvailableGPUs(); in Initialize()
94 if (GetNumAvailableGPUs() > 0) { in TEST_F()
501 if (GetNumAvailableGPUs() > 0) { in TEST_F()
41 gpu_available_ = GetNumAvailableGPUs() > 0; in SetUp()
136 int num_gpus = tensorflow::grappler::GetNumAvailableGPUs(); in TF_NewCluster()
123 const int num_gpus = grappler::GetNumAvailableGPUs(); in ConvertGraphDefToTensorRT()