Searched refs:first_call_function_nn_module_stack (Results 1 – 3 of 3) sorted by relevance
42 def first_call_function_nn_module_stack(graph: torch.fx.Graph) -> Optional[Dict]: function
28 from torch.fx._utils import first_call_function_nn_module_stack
69 from torch.fx._utils import first_call_function_nn_module_stack