Searched refs:rhs_sharding_transposed_to_match_lhs (Results 1 – 1 of 1) sorted by relevance
498 auto rhs_sharding_transposed_to_match_lhs = in PartitionBaseCase() local561 lhs.sharding(), *rhs_sharding_transposed_to_match_lhs) && in PartitionBaseCase()563 lhs.sharding(), *rhs_sharding_transposed_to_match_lhs)) { in PartitionBaseCase()566 auto resharded_lhs = lhs.Reshard(*rhs_sharding_transposed_to_match_lhs); in PartitionBaseCase()711 : &*rhs_sharding_transposed_to_match_lhs; in PartitionBaseCase()1091 : &*rhs_sharding_transposed_to_match_lhs; in PartitionBaseCase()1504 lhs.Reshard(*rhs_sharding_transposed_to_match_lhs).PadWithValue(zero); in PartitionBaseCase()1613 lhs.Reshard(*rhs_sharding_transposed_to_match_lhs).PadWithValue(zero); in PartitionBaseCase()