Searched refs:aligned_output_sharding (Results 1 – 1 of 1) sorted by relevance
119 auto aligned_output_sharding = hlo_sharding_util::TransposeSharding( in PartitionConvolutionWithBatchGroupCount() local126 sharded_conv->set_sharding(aligned_output_sharding); in PartitionConvolutionWithBatchGroupCount()212 auto aligned_output_sharding = hlo_sharding_util::TransposeSharding( in PartitionConvolutionWithFeatureGroupCount() local218 sharded_conv->set_sharding(aligned_output_sharding); in PartitionConvolutionWithFeatureGroupCount()