Searched refs:output_sharding_transposed_to_match_lhs (Results 1 – 1 of 1) sorted by relevance
442 const std::optional<HloSharding>& output_sharding_transposed_to_match_lhs, in GetWindowedEinsumConfiguration() argument626 output_sharding_transposed_to_match_lhs == lhs_sharding &&802 auto output_sharding_transposed_to_match_lhs = in PartitionBaseCase() local936 : &*output_sharding_transposed_to_match_lhs; in PartitionBaseCase()1830 options, output_sharding_transposed_to_match_lhs, in PartitionBaseCase()1912 lhs.Reshard(*output_sharding_transposed_to_match_lhs); in PartitionBaseCase()1923 lhs.Reshard(*output_sharding_transposed_to_match_lhs); in PartitionBaseCase()3108 std::optional<HloSharding> output_sharding_transposed_to_match_lhs = in PrioritizeContractingDimensionsPartitioning() local3132 output_sharding_transposed_to_match_lhs, in PrioritizeContractingDimensionsPartitioning()