Searched defs:per_group_partitioner_state (Results 1 – 3 of 3) sorted by relevance
234 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionGatherIndexPassthroughPartition() local339 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionGatherPassthroughOperand() local411 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionGatherTrivialIndexedOperandDimension() local603 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionGatherIndexParallelDimensions() local914 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionScatterIndexPassthroughPartition() local
2089 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnBatch() local2104 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnBatch() local2402 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnNonContracting() local
1224 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in ReplicatePartial() local1354 auto per_group_partitioner_state = CreatePerGroupPartitioningState( in ReshardToPartialReplicateWithAllGather() local2365 auto per_group_partitioner_state = in HandleConcatenate() local