| 2212 | }; |
| 2213 | |
| 2214 | int GetNumGPUs(const Cluster& cluster) { |
| 2215 | auto devices = cluster.GetDevices(); |
| 2216 | int num_gpus = 0; |
| 2217 | for (const auto& device : devices) { |
| 2218 | if (device.second.type() == "GPU") { |
| 2219 | num_gpus++; |
| 2220 | } |
| 2221 | } |
| 2222 | return num_gpus; |
| 2223 | } |
| 2224 | } // namespace |
| 2225 | |
| 2226 | Status LayoutOptimizer::Tune(const GrapplerItem& item, |