Searched refs:GetInGroupPartitionId (Results 1 – 1 of 1) sorted by relevance
1572 HloInstruction* GetInGroupPartitionId( in GetInGroupPartitionId() function1595 return GetInGroupPartitionId(creator.create_partition_id(b), device_groups, in GetPerGroupCollectiveOpsCreator()1678 GetInGroupPartitionId(state.partition_id, device_groups, b); in CreatePerGroupPartitioningState()