| 234 | } |
| 235 | |
| 236 | void CompositeNodeManager::AddNode(const NodeDef* node) { |
| 237 | if (IsSend(*node)) { |
| 238 | send_manager_.AddNode(node); |
| 239 | } else if (IsRecv(*node)) { |
| 240 | recv_manager_.AddNode(node); |
| 241 | } else { |
| 242 | const auto& device = node_map_->at(node).device_name; |
| 243 | ops_lifo_map_[device].AddNode(node); |
| 244 | } |
| 245 | } |
| 246 | |
| 247 | const NodeDef* CompositeNodeManager::GetCurrNode() { |
| 248 | if (curr_node_) return curr_node_; |