Searched defs:set_current_node (Results 1 – 4 of 4) sorted by relevance
310 set_current_node: Callable[[Any], Any] = _current_node._set_handler variable in _V
1111 def set_current_node(self, node: torch.fx.Node): member in GraphLowering
1870 def set_current_node(node): function
1640 def set_current_node(self, node): member in Kernel