Searched refs:logs_to_string (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_auto_functionalize.py | 13 from torch.testing._internal.logging_utils import logs_to_string 169 log_stream, ctx = logs_to_string( 227 log_stream, ctx = logs_to_string( 316 log_stream, ctx = logs_to_string( 388 log_stream, ctx = logs_to_string( 433 log_stream, ctx = logs_to_string( 449 log_stream, ctx = logs_to_string( 492 log_stream, ctx = logs_to_string( 884 log_stream, ctx = logs_to_string(
|
H A D | test_inplacing_pass.py | 23 from torch.testing._internal.logging_utils import logs_to_string 370 log_stream, ctx = logs_to_string( 396 log_stream, ctx = logs_to_string(
|
H A D | test_compiled_autograd.py | 28 from torch.testing._internal.logging_utils import logs_to_string 2364 logs, ctx = logs_to_string( 2394 logs, ctx = logs_to_string( 2443 logs, ctx = logs_to_string( 2468 logs, ctx = logs_to_string( 2519 logs, ctx = logs_to_string( 2565 logs, ctx = logs_to_string( 2619 logs, ctx = logs_to_string(
|
H A D | test_triton_kernels.py | 21 from torch.testing._internal.logging_utils import logs_to_string 310 log_stream, ctx = logs_to_string("torch._inductor.codecache", "output_code")
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | logging_utils.py | 190 def logs_to_string(module, log_option): function
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_unspec.py | 15 from torch.testing._internal.logging_utils import logs_to_string 668 log_stream, ctx = logs_to_string("torch._dynamo.output_graph", "graph_code")
|
H A D | test_misc.py | 86 from torch.testing._internal.logging_utils import logs_to_string 719 log_stream, ctx = logs_to_string( 778 log_stream, ctx = logs_to_string( 872 log_stream, ctx = logs_to_string(
|