Searched defs:GetGroupMode (Results 1 – 4 of 4) sorted by relevance
167 CollectiveOpGroupMode GetGroupMode(OpT op) { in GetGroupMode() function242 CollectiveOpGroupMode NcclAllReduceThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclAllReduceThunk275 CollectiveOpGroupMode NcclAllReduceStartThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclAllReduceStartThunk346 /*static*/ CollectiveOpGroupMode NcclReduceScatterThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclReduceScatterThunk
52 static CollectiveOpGroupMode GetGroupMode(mlir::lmhlo::AllGatherOp op) { in GetGroupMode() function
52 static CollectiveOpGroupMode GetGroupMode(mlir::lmhlo::AllToAllOp op) { in GetGroupMode() function
71 static CollectiveOpGroupMode GetGroupMode( in GetGroupMode() function