Searched defs:per_group_partitioner_state (Results 1 – 3 of 3) sorted by relevance
417 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionIndexParallelDimensions() local584 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in HandleScatter() local
1727 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnBatch() local1742 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnBatch() local2033 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnNonContracting() local
958 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in ReshardToPartialReplicateWithAllGather() local1559 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in HandleConcatenate() local