Searched defs:node_call_back (Results 1 – 2 of 2) sorted by relevance
485 def sequential_split(gm: torch.fx.GraphModule, node_call_back) -> torch.fx.GraphModule: argument513 def nodes_filter(nodes: List[torch.fx.Node], node_call_back) -> List[torch.fx.Node]: argument519 nodes: List[torch.fx.Node], node_call_back=None argument531 def nodes_count(nodes: List[torch.fx.Node], node_call_back) -> int: argument536 def nodes_map(nodes: List[torch.fx.Node], node_call_back) -> List[torch.fx.Node]: argument
115 def node_call_back(node: torch.fx.Node): function