Searched refs:inner_state (Results 1 – 1 of 1) sorted by relevance
2220 auto inner_state = CreatePerGroupPartitioningState( in PartitionDotGroupOnContracting() local2227 inner_state), in PartitionDotGroupOnContracting()2230 inner_state), in PartitionDotGroupOnContracting()2763 auto inner_state = CreatePerGroupPartitioningState( in PartitionDot() local2767 PartitionDot(PartitionedHlo(lhs.hlo(), lhs.base_shape(), inner_state), in PartitionDot()2768 PartitionedHlo(rhs.hlo(), rhs.base_shape(), inner_state), in PartitionDot()