Searched defs:aligned_output_sharding (Results 1 – 1 of 1) sorted by relevance
119 auto aligned_output_sharding = hlo_sharding_util::TransposeSharding( in PartitionConvolutionWithBatchGroupCount() local212 auto aligned_output_sharding = hlo_sharding_util::TransposeSharding( in PartitionConvolutionWithFeatureGroupCount() local