| 136 | } |
| 137 | |
| 138 | Status HeapReadyManager::Init( |
| 139 | const std::unordered_map<const NodeDef*, NodeState>* node_map) { |
| 140 | // Resets the node state since different instances of the scheduler can reuse |
| 141 | // the same node_manager. |
| 142 | node_map_ = node_map; |
| 143 | nodes_.clear(); |
| 144 | waiting_queue_.clear(); |
| 145 | |
| 146 | // Sets up the comparator for the heap. |
| 147 | greater_ = Greater(); |
| 148 | |
| 149 | return Status::OK(); |
| 150 | } |
| 151 | |
| 152 | const NodeDef* HeapReadyManager::GetCurrNode() { |
| 153 | if (nodes_.empty()) { |