Searched refs:extract_compiled_graph (Results 1 – 3 of 3) sorted by relevance
18 from torch._lazy.extract_compiled_graph import extract_compiled_graph139 optimized_mod = extract_compiled_graph(fx.symbolic_trace(mod), args)
37 compiled_graph = bridge.extract_compiled_graph(model, args)
152 def extract_compiled_graph(model: fx.GraphModule, example_inputs) -> Callable: function