Home
last modified time | relevance | path

Searched refs:micro_pipeline_tp_pass (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/
H A Dtest_micro_pipeline_tp.py13 micro_pipeline_tp_pass,
277 micro_pipeline_tp_pass(gm.graph)
369 micro_pipeline_tp_pass(gm.graph)
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpost_grad.py50 from .micro_pipeline_tp import micro_pipeline_tp_pass
118 micro_pipeline_tp_pass(gm.graph)
H A Dmicro_pipeline_tp.py834 def micro_pipeline_tp_pass(graph: torch.fx.Graph): function