Searched refs:log_url_for_graph_xform (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
H A D | pre_grad.py | 258 gm, "pre_grad_custom_pass", config.trace.log_url_for_graph_xform 302 gm, "linear_permute_fusion", config.trace.log_url_for_graph_xform 306 gm, "permute_linear_fusion", config.trace.log_url_for_graph_xform 310 gm, "permute_matmul_fusion", config.trace.log_url_for_graph_xform 319 gm, "remove_identity", config.trace.log_url_for_graph_xform 323 gm, "fuse_conv_bn", config.trace.log_url_for_graph_xform
|
H A D | joint_graph.py | 443 graph, "joint_custom_pre_pass", config.trace.log_url_for_graph_xform 454 graph, "constant_fold_uniform_value", config.trace.log_url_for_graph_xform 467 graph, "joint_custom_post_pass", config.trace.log_url_for_graph_xform
|
H A D | replace_random.py | 31 gm, "fuse_seed_creation_pass", config.trace.log_url_for_graph_xform
|
H A D | post_grad.py | 90 gm, "post_grad_custom_pre_pass", config.trace.log_url_for_graph_xform 129 gm, "post_grad_custom_post_pass", config.trace.log_url_for_graph_xform
|
H A D | ddp_fusion.py | 588 config.trace.log_url_for_graph_xform,
|
H A D | group_batch_fusion.py | 1315 config.trace.log_url_for_graph_xform,
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_graph_transform_observer.py | 48 inductor_config.trace.log_url_for_graph_xform = log_url
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | config.py | 1203 log_url_for_graph_xform = os.environ.get("INDUCTOR_LOG_URL_FOR_GRAPH_XFORM", None) variable in trace
|
H A D | utils.py | 1614 with GraphTransformObserver(gm, msg, config.trace.log_url_for_graph_xform):
|
H A D | pattern_matcher.py | 1703 gm, pass_name, trace_config.log_url_for_graph_xform
|