_ReduceScatterMatch (definition) in projects: external

Searched defs:_ReduceScatterMatch (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/_inductor/fx_passes/
D	micro_pipeline_tp.py	`200 class _ReduceScatterMatch: class 668 def fuse_matmul_reduce_scatter(reduce_scatter: _ReduceScatterMatch) -> None:`