Home
last modified time | relevance | path

Searched refs:set_current_node (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/_inductor/
Dvirtualized.py310 set_current_node: Callable[[Any], Any] = _current_node._set_handler variable in _V
Dgraph.py1111 def set_current_node(self, node: torch.fx.Node): member in GraphLowering
1184 with ir.IRNode.current_origins(origins), self.set_current_node(
1186 ), V.set_current_node(n):
Dscheduler.py667 ) as flop_counter_mode, V.set_current_node(
988 ), V.kernel.set_current_node(self):
/external/pytorch/torch/_dynamo/
Dutils.py1870 def set_current_node(node): function
1896 with set_current_node(node):
/external/pytorch/torch/_inductor/codegen/
Dcommon.py1640 def set_current_node(self, node): member in Kernel