Home
last modified time | relevance | path

Searched refs:GetNumGPUs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer.cc49 inline std::pair<int, int> GetNumGPUs(const Cluster& cluster) { in GetNumGPUs() function
411 const auto num_gpus_and_num_volta = GetNumGPUs(*cluster); in Optimize()
Dauto_mixed_precision.cc1967 int GetNumGPUs(const Cluster& cluster, in GetNumGPUs() function
2004 int num_gpus = ShouldIgnorePerformance() ? GetNumGPUs(*cluster) in Optimize()
2005 : GetNumGPUs(*cluster, kMinGPUArch); in Optimize()
Dlayout_optimizer.cc2223 int GetNumGPUs(const Cluster& cluster) { in GetNumGPUs() function
2259 if (GetNumGPUs(*cluster) < 1) { in Optimize()