Searched refs:_aot_export_function (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 1246 fx_g, metadata, in_spec, out_spec = _aot_export_function(
1351 fx_g, metadata, in_spec, out_spec = _aot_export_function(
1416 def _aot_export_function( function
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_aot_autograd.py | 14 from torch._functorch.aot_autograd import _aot_export_function, create_functional_call
834 fx_g, _, _, _ = _aot_export_function(
|