Searched refs:rhs_sharding_transposed_to_match_lhs (Results 1 – 1 of 1) sorted by relevance
445 const std::optional<HloSharding>& rhs_sharding_transposed_to_match_lhs, in GetWindowedEinsumConfiguration() argument572 rhs_sharding_transposed_to_match_lhs.has_value() && in __anon991ccdc40802()575 new_lhs = new_lhs.Reshard(*rhs_sharding_transposed_to_match_lhs); in __anon991ccdc40802()790 auto rhs_sharding_transposed_to_match_lhs = in PartitionBaseCase() local853 lhs.sharding(), *rhs_sharding_transposed_to_match_lhs) && in PartitionBaseCase()855 lhs.sharding(), *rhs_sharding_transposed_to_match_lhs)) { in PartitionBaseCase()858 auto resharded_lhs = lhs.Reshard(*rhs_sharding_transposed_to_match_lhs); in PartitionBaseCase()921 rhs_sharding_transposed_to_match_lhs.has_value() && in PartitionBaseCase()925 lhs = lhs.Reshard(*rhs_sharding_transposed_to_match_lhs).PadWithZero(); in PartitionBaseCase()940 : &*rhs_sharding_transposed_to_match_lhs; in PartitionBaseCase()[all …]