Home
last modified time | relevance | path

Searched refs:GetNumGPUs (Results 1 – 2 of 2) 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
425 const auto num_gpus_and_num_volta = GetNumGPUs(*cluster); in Optimize()
Dauto_mixed_precision.cc2253 int GetNumGPUs(const Cluster& cluster) { in GetNumGPUs() function
2292 int num_gpus = GetNumGPUs(*cluster); in Optimize()