| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_minifier.py | 21 def has_mul(fx_g, inps): argument
41 def has_nans(fx_g, inps): argument
61 def inputs_returned(fx_g, inps): argument
78 def has_add(fx_g, inps): argument
105 def pass_checker(fx_g, inps): argument
|
| H A D | test_aotdispatch.py | 3168 def get_graph_size(fx_g, _): argument
3252 def compiler(fx_g, _): argument
3983 def extract_graph(fx_g, _, graph_cell): argument
3988 def get_ins_outs(fx_g): argument
3999 def get_num_ins_outs(fx_g): argument
5084 def check_meta_tensor(fx_g, _): argument
|
| H A D | test_eager_transforms.py | 4588 def normalize_devices(fx_g): argument
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | compilers.py | 36 def _canonicalize(fx_g): argument
101 def _draw_graph_compile(fx_g, _, name, clear_meta=True): argument
190 def simple_ts_compile(fx_g, _): argument
231 def print_compile(fx_g, _): argument
275 def debug_compile(fx_g, inps): argument
|
| H A D | fx_minifier.py | 147 def dump_state(fx_g, inps): argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/repro/ |
| H A D | after_aot.py | 344 fx_g, argument
408 def inductor_fails(fx_g, args, check_str=None): argument
444 fx_g, args, check_str=None, *, require_fp64=False, ignore_non_fp=False argument
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | utils.py | 106 def f(fx_g, inps): argument
400 def copy_fwd_metadata_to_bw_nodes(fx_g): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/ |
| H A D | _op_coverage.py | 19 def fwd_bwd_compiler(fx_g, _): argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | debugging.py | 91 def boxed_nop(fx_g, example_inputs): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_compile.py | 63 def extract_graph(fx_g, _, graph_cell): argument
|
| /aosp_15_r20/external/pytorch/test/higher_order_ops/ |
| H A D | test_with_effects.py | 47 def extract_graph(fx_g, _, graph_cell): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_proxy_tensor.py | 926 def _get_node(fx_g, cond): argument
1872 def _assert_no_guards(self, fx_g, free_symbols): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_aot_autograd.py | 743 def test_compile(fx_g, example_inps): argument
|