Home
last modified time | relevance | path

Searched refs:aot_export_module (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/test/higher_order_ops/
Dtest_with_effects.py19 from torch._functorch.aot_autograd import aot_export_module
109 gm, gs = aot_export_module(M(), inputs, trace_joint=False)
125 gm, gs = aot_export_module(M(), inputs, trace_joint=False)
150 gm, gs = aot_export_module(M(), (torch.ones(2, 3),), trace_joint=False)
183 gm, gs = aot_export_module(M(), inputs, trace_joint=False)
217 gm, gs = aot_export_module(M(), inputs, trace_joint=False)
/external/pytorch/test/functorch/
Dtest_aotdispatch.py44 from torch._functorch.aot_autograd import aot_export_joint_simple, aot_export_module
4042 aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4044 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=False)
4088 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4113 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4135 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4157 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4184 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4208 gm, _ = aot_export_module(mod, [inp], trace_joint=False, pre_dispatch=True)
4234 gm, graph_sig = aot_export_module(
[all …]
/external/pytorch/test/export/
Dtest_experimental.py10 from torch._functorch.aot_autograd import aot_export_module
Dtest_torchbind.py9 from torch._functorch.aot_autograd import aot_export_module
961 gm = aot_export_module(mod, (fake_tq1, fake_x), trace_joint=False)[0]
/external/pytorch/torch/export/
Dexported_program.py298 from torch._functorch.aot_autograd import aot_export_module
410 gm, graph_signature = aot_export_module(
D_trace.py58 from torch._functorch.aot_autograd import aot_export_module
637 gm, graph_signature = transform(aot_export_module)(
/external/pytorch/torch/_inductor/
Dcompile_fx.py34 from torch._functorch.aot_autograd import aot_export_module, make_boxed_func
1487 gm, graph_signature = aot_export_module(
/external/pytorch/torch/_functorch/
Daot_autograd.py1110 def aot_export_module( function