Home
last modified time | relevance | path

Searched refs:fx_graph_cache (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_autoheuristic.py146 fx_graph_cache=False,
H A Dtest_pattern_matcher.py307 fx_graph_cache=False,
367 fx_graph_cache=False,
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dtimm_models.py24 torch._inductor.config.fx_graph_cache = True
H A Dtorchbench.py30 torch._inductor.config.fx_graph_cache = True
H A Dhuggingface.py38 torch._inductor.config.fx_graph_cache = True
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Dautograd_cache.py145 if not torch._inductor.config.fx_graph_cache:
/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A Dgenerate.py21 torch._inductor.config.fx_graph_cache = True # Experimental feature to reduce compilation times, w…
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dconfig.py43 fx_graph_cache = ( variable
H A Dcompile_fx.py514 and (config.fx_graph_cache or fx_graph_remote_cache)
531 local=config.fx_graph_cache,