Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/_tools/
H A Dfsdp2_mem_tracker.py497 op: ReduceOp.RedOpType = dist.ReduceOp.SUM,
519 op: ReduceOp.RedOpType = dist.ReduceOp.SUM,
/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D_fsdp_collectives.py289 reduce_scatter_reduce_op: Optional[Union[dist.ReduceOp, dist.ReduceOp.RedOpType]],
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DTypes.hpp36 enum RedOpType : uint8_t { enum