Searched refs:set_current_node (Results 1 – 5 of 5) 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 GraphLowering1184 with ir.IRNode.current_origins(origins), self.set_current_node(1186 ), V.set_current_node(n):
667 ) as flop_counter_mode, V.set_current_node(988 ), V.kernel.set_current_node(self):
1870 def set_current_node(node): function1896 with set_current_node(node):
1640 def set_current_node(self, node): member in Kernel