Home
last modified time | relevance | path

Searched refs:extract_compiled_graph (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/lazy/
H A Dtest_extract_compiled_graph.py18 from torch._lazy.extract_compiled_graph import extract_compiled_graph
139 optimized_mod = extract_compiled_graph(fx.symbolic_trace(mod), args)
/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Dtorchxla.py37 compiled_graph = bridge.extract_compiled_graph(model, args)
/aosp_15_r20/external/pytorch/torch/_lazy/
H A Dextract_compiled_graph.py152 def extract_compiled_graph(model: fx.GraphModule, example_inputs) -> Callable: function