Searched refs:lhs_sharding_transposed_to_match_rhs (Results 1 – 1 of 1) sorted by relevance
494 auto lhs_sharding_transposed_to_match_rhs = in PartitionBaseCase() local523 lhs_sharding_transposed_to_match_rhs == rhs_sharding) { in PartitionBaseCase()544 rhs.sharding(), *lhs_sharding_transposed_to_match_rhs) && in PartitionBaseCase()546 rhs.sharding(), *lhs_sharding_transposed_to_match_rhs)) { in PartitionBaseCase()549 auto resharded_rhs = rhs.Reshard(*lhs_sharding_transposed_to_match_rhs); in PartitionBaseCase()710 ? &*lhs_sharding_transposed_to_match_rhs in PartitionBaseCase()1090 ? &*lhs_sharding_transposed_to_match_rhs in PartitionBaseCase()1509 rhs.Reshard(*lhs_sharding_transposed_to_match_rhs).PadWithValue(zero); in PartitionBaseCase()1618 rhs.Reshard(*lhs_sharding_transposed_to_match_rhs).PadWithValue(zero); in PartitionBaseCase()