| 250 | } |
| 251 | |
| 252 | int GetGpuNumber(const string& device_name) { |
| 253 | DeviceNameUtils::ParsedName parsed_name; |
| 254 | if (!DeviceNameUtils::ParseFullName(device_name, &parsed_name)) { |
| 255 | return -1; |
| 256 | } |
| 257 | |
| 258 | return parsed_name.type == DEVICE_GPU ? parsed_name.id : -1; |
| 259 | } |
| 260 | } // namespace |
| 261 | |
| 262 | bool IsSingleGpuGraph(const Graph& g) { |