Searched defs:group_mode (Results 1 – 6 of 6) sorted by relevance
115 CollectiveOpGroupMode group_mode) { in CollectiveOpGroupModeToString()131 CollectiveOpGroupMode group_mode) { in GetParticipatingDevicesGroups()240 CollectiveOpGroupMode group_mode) { in GetParticipatingDevices()
142 StatusOr<CollectiveOpGroupMode> group_mode = in ToString() local396 StatusOr<CollectiveOpGroupMode> group_mode = in TEST_P() local
34 CollectiveOpGroupMode group_mode, in CreateStartIndicesForCollectiveDecomposition()
86 CollectiveOpGroupMode group_mode) { in GetSubgroupSize()234 CollectiveOpGroupMode group_mode, in CheckReplicaGroups()607 CollectiveOpGroupMode group_mode) { in CheckDuplicatedSourceOrTarget()
74 CollectiveOpGroupMode group_mode; member
3003 CollectiveOpGroupMode group_mode = NcclThunkType::GetGroupMode(op); in EmitNcclThunk() local