Searched refs:assume_lhs_match (Results 1 – 1 of 1) sorted by relevance
2316 [&](bool assume_lhs_match) -> absl::optional<int64> { in LhsIsBestMatchForNonContractingPartitioning() argument2319 assume_lhs_match ? dims_mapping.lhs_non_contracting_dims in LhsIsBestMatchForNonContractingPartitioning()2323 assume_lhs_match ? dims_mapping.rhs_non_contracting_dims in LhsIsBestMatchForNonContractingPartitioning()2326 assume_lhs_match ? indices_map.output_to_lhs_indices in LhsIsBestMatchForNonContractingPartitioning()2329 assume_lhs_match ? indices_map.output_to_rhs_indices in LhsIsBestMatchForNonContractingPartitioning()2332 assume_lhs_match ? indices_map.lhs_to_output_indices in LhsIsBestMatchForNonContractingPartitioning()2335 assume_lhs_match ? indices_map.rhs_to_output_indices in LhsIsBestMatchForNonContractingPartitioning()2338 assume_lhs_match ? lhs.sharding() : rhs.sharding(); in LhsIsBestMatchForNonContractingPartitioning()2340 assume_lhs_match ? rhs.sharding() : lhs.sharding(); in LhsIsBestMatchForNonContractingPartitioning()2341 const PartitionedHlo& matching_partitioned = assume_lhs_match ? lhs : rhs; in LhsIsBestMatchForNonContractingPartitioning()[all …]