Searched defs:example_inputs_ (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_backward_higher_order_ops.py | 98 def inner_compiler(gm_, example_inputs_): argument
155 def inner_compiler(gm_, example_inputs_): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_codegen_dynamic_shapes.py | 67 def compile_fx_wrapper(model_, example_inputs_): argument
|
| H A D | test_compiled_autograd.py | 40 def _inner_compiler(gm_, example_inputs_): argument
445 def inner_compiler(gm_, example_inputs_): argument
1133 def inner_compiler(gm_, example_inputs_): argument
|
| H A D | test_cpu_repro.py | 3032 def compile_fx_wrapper(model_, example_inputs_): argument
|
| H A D | test_torchinductor.py | 435 def compile_fx_wrapper(model_, example_inputs_): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/ |
| H A D | test_replicate_with_compiler.py | 62 def inner_compiler(gm_, example_inputs_): argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | utils.py | 2663 def inner_compiler(gm_, example_inputs_): argument
|