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::NcclAllReduceThunk279 CollectiveOpGroupMode NcclAllReduceStartThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclAllReduceStartThunk354 /*static*/ CollectiveOpGroupMode NcclReduceScatterThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclReduceScatterThunk
51 static CollectiveOpGroupMode GetGroupMode(mlir::lmhlo::AllGatherOp op) { in GetGroupMode() function
51 static CollectiveOpGroupMode GetGroupMode(mlir::lmhlo::AllToAllOp op) { in GetGroupMode() function
78 static CollectiveOpGroupMode GetGroupMode( in GetGroupMode() function