Searched refs:lhs_sharding_transposed_to_match_rhs (Results 1 – 1 of 1) sorted by relevance
444 const std::optional<HloSharding>& lhs_sharding_transposed_to_match_rhs, in GetWindowedEinsumConfiguration() argument573 lhs_sharding_transposed_to_match_rhs.has_value()) { in __anon991ccdc40802()577 new_rhs = new_rhs.Reshard(*lhs_sharding_transposed_to_match_rhs); in __anon991ccdc40802()786 auto lhs_sharding_transposed_to_match_rhs = in PartitionBaseCase() local815 lhs_sharding_transposed_to_match_rhs == rhs_sharding) { in PartitionBaseCase()836 rhs.sharding(), *lhs_sharding_transposed_to_match_rhs) && in PartitionBaseCase()838 rhs.sharding(), *lhs_sharding_transposed_to_match_rhs)) { in PartitionBaseCase()841 auto resharded_rhs = rhs.Reshard(*lhs_sharding_transposed_to_match_rhs); in PartitionBaseCase()922 lhs_sharding_transposed_to_match_rhs.has_value()) { in PartitionBaseCase()927 rhs = rhs.Reshard(*lhs_sharding_transposed_to_match_rhs).PadWithZero(); in PartitionBaseCase()[all …]