| 48 | } |
| 49 | |
| 50 | void LogDeviceAssignment(const Node* node, bool log_device_placement) { |
| 51 | // Log placement if log_device_placement is set. |
| 52 | if (log_device_placement) { |
| 53 | printf("%s: (%s): %s\n", node->name().c_str(), node->type_string().c_str(), |
| 54 | node->assigned_device_name().c_str()); |
| 55 | LOG(INFO) << node->name() << ": " |
| 56 | << "(" << node->type_string() |
| 57 | << "): " << node->assigned_device_name(); |
| 58 | } |
| 59 | } |
| 60 | |
| 61 | Status AssignAndLog(int assigned_device, Node* node, |
| 62 | ColocationGraph* colocation_graph, |
no test coverage detected