| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_backward_higher_order_ops.py | 97 def compiler_fn(gm): function
154 def compiler_fn(gm): function
228 def compiler_fn(gm): function
|
| H A D | test_backends.py | 52 def compiler_fn(graph, example_inputs): function
80 def compiler_fn(graph, example_inputs): function
|
| H A D | test_verify_correctness.py | 68 def compiler_fn(graph, example_inputs): function
|
| H A D | test_hooks.py | 17 def compiler_fn(gm): function
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/repro/ |
| H A D | after_dynamo.py | 50 def _accuracy_fails(gm, example_inputs, compiler_fn): argument
345 def backend_fails(gm, example_inputs, compiler_fn, orig_failure): argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | compiled_autograd.py | 62 def __init__(self, compiler_fn) -> None: argument
498 def enable(compiler_fn): argument
|
| H A D | debug_utils.py | 440 compiler_fn, argument
|
| H A D | eval_frame.py | 668 def get_compiler_fn(compiler_fn): argument
|
| H A D | utils.py | 2662 def compiler_fn(gm): function
|
| H A D | symbolic_convert.py | 2656 compiler_fn, argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | registry.py | 62 def lookup_backend(compiler_fn): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/ |
| H A D | test_replicate_with_compiler.py | 60 def compiler_fn(no_inductor=False): function
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_compile.py | 327 def compiler_fn(compiled_autograd_backend): function
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_autograd.py | 54 compiler_fn = make_compiler_fn() variable
84 self, fn, count=1, compiler_fn=compiler_fn, compile_fn=False argument
|