Searched refs:track_graph_compiling (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
H A D | jit_compile_runtime_wrappers.py | 41 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 D | logging_utils.py | 47 def track_graph_compiling(aot_config, graph_name): function
|
H A D | runtime_wrappers.py | 42 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 D | aot_autograd.py | 76 track_graph_compiling,
|