Searched refs:grouped_state (Results 1 – 1 of 1) sorted by relevance
2772 auto grouped_state = CreatePerGroupPartitioningState( in HandleReduce() local2791 PartitionedHlo(reshape, all_gather_shape, grouped_state) in HandleReduce()