Searched refs:per_group_state (Results 1 – 1 of 1) sorted by relevance
3087 auto per_group_state = CreatePerGroupPartitioningState( in HandleRng() local3094 return PartitionedHlo(rng, rng->shape(), per_group_state) in HandleRng()