Home
last modified time | relevance | path

Searched refs:root_module_when_exporting_non_strict (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Ddispatch_and_compile_graph.py37 root_module_when_exporting_non_strict,
123 mod_when_exporting_non_strict = root_module_when_exporting_non_strict(flat_fn)
H A Dutils.py390 def root_module_when_exporting_non_strict(flat_fn): function
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py125 root_module_when_exporting_non_strict,
617 mod = root_module_when_exporting_non_strict(flat_fn)