Home
last modified time | relevance | path

Searched refs:select_decomp_table (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/
H A D_op_coverage.py10 from torch._inductor.decomposition import select_decomp_table
14 inductor_decomps = select_decomp_table()
/aosp_15_r20/external/pytorch/torch/_export/
H A D__init__.py266 from torch._inductor.decomposition import select_decomp_table
/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Ddebugging.py134 ).select_decomp_table(),
/aosp_15_r20/external/pytorch/test/
H A Dtest_out_dtype_op.py191 decomp_table = torch._inductor.decomposition.select_decomp_table()
H A Dtest_decomp.py612 decomp_table = torch._inductor.decomposition.select_decomp_table()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dpattern_matcher.py95 from .decomposition import select_decomp_table
1832 get_decomp_fn() if get_decomp_fn is not None else select_decomp_table()
1864 decompositions=select_decomp_table(),
H A Dcompile_fx.py62 from .decomposition import select_decomp_table
1349 decompositions if decompositions is not None else select_decomp_table()
H A Ddecomposition.py790 def select_decomp_table() -> Dict[Any, Callable[..., Any]]: function
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_activation_checkpointing.py367 ).select_decomp_table(),
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py10578 from torch._inductor.decomposition import select_decomp_table
10590 fw_module = make_fx(interpret, select_decomp_table())(