Searched refs:CreatePerGroupPartitioningState (Results 1 – 5 of 5) sorted by relevance
346 PartitionedHlo::PartitioningState CreatePerGroupPartitioningState(
417 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionIndexParallelDimensions()584 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in HandleScatter()
1727 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnBatch()1742 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnBatch()2033 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnNonContracting()2220 auto inner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnContracting()2763 auto inner_state = CreatePerGroupPartitioningState( in PartitionDot()
958 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in ReshardToPartialReplicateWithAllGather()1559 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in HandleConcatenate()2772 auto grouped_state = CreatePerGroupPartitioningState( in HandleReduce()3087 auto per_group_state = CreatePerGroupPartitioningState( in HandleRng()
1671 PartitionedHlo::PartitioningState CreatePerGroupPartitioningState( in CreatePerGroupPartitioningState() function