| 462 | } |
| 463 | |
| 464 | void OpGraph::ForEachDataAndCtrlInNode(OpNode* node, |
| 465 | const std::function<void(OpNode*)>& Handler) const { |
| 466 | node->ForEachNodeOnInEdge(Handler); |
| 467 | for (const auto& ctrl_in_op_name : node->op().op_conf().ctrl_in_op_name()) { |
| 468 | CHECK(op_name2op_node_.find(ctrl_in_op_name) != op_name2op_node_.end()) |
| 469 | << " cannot find ctrl_in_op_name: [" << ctrl_in_op_name << "] of op: [" |
| 470 | << node->op().op_name() << "] in OpGraph. "; |
| 471 | Handler(op_name2op_node_.at(ctrl_in_op_name)); |
| 472 | } |
| 473 | } |
| 474 | |
| 475 | void OpGraph::ForEachDataAndCtrlOutNode(OpNode* node, |
| 476 | const std::function<void(OpNode*)>& Handler) const { |
no test coverage detected