Searched refs:lazy_format_graph_code (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | utils.py | 348 from torch._dynamo.utils import lazy_format_graph_code
352 lazy_format_graph_code(
369 from torch._dynamo.utils import lazy_format_graph_code
373 lazy_format_graph_code(
|
| H A D | jit_compile_runtime_wrappers.py | 21 from torch._dynamo.utils import lazy_format_graph_code
343 lazy_format_graph_code(
525 lazy_format_graph_code(
536 lazy_format_graph_code(
|
| H A D | dispatch_and_compile_graph.py | 15 from torch._dynamo.utils import lazy_format_graph_code
206 lazy_format_graph_code(
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | split_module.py | 11 from torch.fx._utils import lazy_format_graph_code
149 lazy_format_graph_code(
571 lazy_format_graph_code(
|
| H A D | runtime_assert.py | 22 from torch.fx._utils import lazy_format_graph_code
116 lazy_format_graph_code(
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | freezing.py | 11 from torch._dynamo.utils import dynamo_timed, lazy_format_graph_code
118 "%s", lazy_format_graph_code("FROZEN GRAPH", aot_autograd_gm, colored=True)
|
| H A D | compile_fx.py | 31 lazy_format_graph_code,
758 lazy_format_graph_code(
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | _utils.py | 9 def lazy_format_graph_code(name, gm, maybe_id=None, **kwargs): function
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | compiled_autograd.py | 13 from torch._dynamo.utils import counters, lazy_format_graph_code, set_locals_to_steal
319 lazy_graph_code = lazy_format_graph_code(
|
| H A D | output_graph.py | 84 lazy_format_graph_code,
1342 lazy_format_graph_code(
|
| H A D | utils.py | 100 from torch.fx._utils import _format_graph_code, lazy_format_graph_code
2082 + str(lazy_format_graph_code(fn_name, gm))
|