MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / safe_remove_nodes

Method safe_remove_nodes

python/paddle/base/framework.py:5837–5851  ·  view source on GitHub ↗

Remove nodes safely since links connected to these removed nodes are also removed. Args: remove_nodes(set): the nodes prepared to be removed.

(self, remove_nodes)

Source from the content-addressed store, hash-verified

5835 node_out.append_input(node_in)
5836
5837 def safe_remove_nodes(self, remove_nodes):
5838 """
5839 Remove nodes safely since links connected to these removed nodes are
5840 also removed.
5841
5842 Args:
5843 remove_nodes(set): the nodes prepared to be removed.
5844 """
5845 if not isinstance(remove_nodes, set):
5846 if isinstance(remove_nodes, Iterable):
5847 remove_nodes = set(remove_nodes)
5848 else:
5849 remove_nodes = {remove_nodes}
5850 original_nodes = {n.node for n in remove_nodes}
5851 core.graph_safe_remove_nodes(self.graph, original_nodes)
5852
5853 def resolve_hazard(self):
5854 ordered_nodes = core.topology_sort(self.graph)

Callers 15

drawMethod · 0.95
_insert_funcMethod · 0.80
applyMethod · 0.80
applyMethod · 0.80
applyMethod · 0.80
applyMethod · 0.80
_remove_fake_quantizeMethod · 0.80

Calls 1

setClass · 0.85

Tested by

no test coverage detected