Searched refs:_extract_graph_module_outputs (Results 1 – 1 of 1) sorted by relevance
273 def _extract_graph_module_outputs(graph_module: torch.fx.GraphModule) -> Any: function286 _extract_graph_module_outputs(graph_module)926 extracted_outputs = _extract_graph_module_outputs(graph_module)