Searched refs:output_sharding_transposed_to_match_rhs (Results 1 – 1 of 1) sorted by relevance
443 const std::optional<HloSharding>& output_sharding_transposed_to_match_rhs, in GetWindowedEinsumConfiguration() argument654 output_sharding_transposed_to_match_rhs == rhs_sharding &&806 auto output_sharding_transposed_to_match_rhs = in PartitionBaseCase() local935 ? &*output_sharding_transposed_to_match_rhs in PartitionBaseCase()1831 output_sharding_transposed_to_match_rhs, in PartitionBaseCase()1914 rhs.Reshard(*output_sharding_transposed_to_match_rhs); in PartitionBaseCase()1934 rhs.Reshard(*output_sharding_transposed_to_match_rhs); in PartitionBaseCase()3112 std::optional<HloSharding> output_sharding_transposed_to_match_rhs = in PrioritizeContractingDimensionsPartitioning() local3133 output_sharding_transposed_to_match_rhs, in PrioritizeContractingDimensionsPartitioning()