Searched refs:create_functional_call (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_functorch/ |
| D | aot_autograd.py | 107 create_functional_call,
1054 functional_call = create_functional_call(mod, params_spec, params_len)
1171 functional_call = create_functional_call(
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | traced_function_transforms.py | 844 def create_functional_call(mod, params_spec, params_len, store_orig_mod=False): function
|
| /external/pytorch/test/dynamo/ |
| D | test_aot_autograd.py | 14 from torch._functorch.aot_autograd import _aot_export_function, create_functional_call
824 functional_call = create_functional_call(g_mod, params_spec, params_len)
|
| /external/pytorch/torch/export/ |
| D | _trace.py | 55 create_functional_call,
1416 functional_call = create_functional_call(
|