Searched refs:output_sharding_transposed_to_match_rhs (Results 1 – 1 of 1) sorted by relevance
392 const absl::optional<HloSharding>& output_sharding_transposed_to_match_rhs, in GetWindowedEinsumConfiguration() argument420 output_sharding_transposed_to_match_rhs == rhs_sharding && in GetWindowedEinsumConfiguration()514 auto output_sharding_transposed_to_match_rhs = in PartitionBaseCase() local706 ? &*output_sharding_transposed_to_match_rhs in PartitionBaseCase()1086 ? &*output_sharding_transposed_to_match_rhs in PartitionBaseCase()1479 output_sharding_transposed_to_match_rhs, lhs_sharding, rhs_sharding); in PartitionBaseCase()1554 rhs.Reshard(*output_sharding_transposed_to_match_rhs); in PartitionBaseCase()1574 rhs.Reshard(*output_sharding_transposed_to_match_rhs); in PartitionBaseCase()