Searched defs:aligned_lhs_sharding (Results 1 – 1 of 1) sorted by relevance
98 auto aligned_lhs_sharding = in PartitionConvolutionWithBatchGroupCount() local189 auto aligned_lhs_sharding = in PartitionConvolutionWithFeatureGroupCount() local255 auto aligned_lhs_sharding = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local565 auto aligned_lhs_sharding = hlo_sharding_util::TransposeSharding( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local