Searched refs:inner_output_sharding (Results 1 – 1 of 1) sorted by relevance
2174 HloSharding inner_output_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local2188 inner_output_sharding = std::move(grouped.sharding); in PartitionDotGroupOnContracting()2214 inner_output_sharding = grouped.sharding; in PartitionDotGroupOnContracting()2231 output_base_shape, inner_output_sharding, dims_mapping, in PartitionDotGroupOnContracting()