Method add_node

python/paddle/distributed/auto_parallel/static/graph.py:118–129

(self, node_id, **attrs)

        return self._adjs

    def add_node(self, node_id, **attrs):
        if node_id is None:
            raise ValueError("None cannot be a node")
        if node_id not in self._nodes:
            node = Node(node_id, **attrs)
            self._nodes[node_id] = node
            self._adjs[node_id] = {}
            self._reverse_adjs[node_id] = []
        else:
            self._nodes[node_id].attrs.update(attrs)

        return self._nodes[node_id]

    def add_edge(self, src_id, tgt_id, **attrs):
        # add nodes