Searched refs:node_replace_ (Results 1 – 2 of 2) sorted by relevance
11 from ..utils import node_replace_, nodes_map62 node_replace_(node, call_func_node)92 node_replace_(node, get_item_node)
546 def node_replace_(old_node: torch.fx.Node, new_node: torch.fx.Node) -> None: function574 node_replace_(ph, arg)579 node_replace_(node, new_node)589 node_replace_(call_mod_node, new_output)604 lambda get_item_node: node_replace_(