Searched defs:aligned_rhs_sharding (Results 1 – 1 of 1) sorted by relevance
96 auto aligned_rhs_sharding = in PartitionConvolutionWithBatchGroupCount() local187 auto aligned_rhs_sharding = in PartitionConvolutionWithFeatureGroupCount() local253 auto aligned_rhs_sharding = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local562 auto aligned_rhs_sharding = hlo_sharding_util::ReverseSharding( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local