Method ForEachNodeOnInEdge

oneflow/core/graph/node.h:95–97

  const std::unordered_set<EdgeType*>& out_edges() const { return out_edges_; }

  void ForEachNodeOnInEdge(std::function<void(NodeType*)> Handler) const {
    for (EdgeType* edge : in_edges_) { Handler(edge->src_node()); }
  }
  void ForEachNodeOnOutEdge(std::function<void(NodeType*)> Handler) const {
    for (EdgeType* edge : out_edges_) { Handler(edge->dst_node()); }
  }