Home
last modified time | relevance | path

Searched refs:track_graph_compiling (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Djit_compile_runtime_wrappers.py41 from .logging_utils import track_graph_compiling
164 with context(), track_graph_compiling(aot_config, "inference"):
365 with track_graph_compiling(aot_config, "joint"):
560 with track_graph_compiling(aot_config, "forward"), torch._C._DisableAutocast():
625 with track_graph_compiling(aot_config, "backward"), torch._C._DisableAutocast():
H A Dlogging_utils.py47 def track_graph_compiling(aot_config, graph_name): function
H A Druntime_wrappers.py42 from .logging_utils import describe_input, format_guard_bug_msg, track_graph_compiling
1953 ), context(), track_graph_compiling(aot_config, "backward"):
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py76 track_graph_compiling,