Home
last modified time | relevance | path

Searched refs:assume_lhs_match (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2316 [&](bool assume_lhs_match) -> absl::optional<int64> { in LhsIsBestMatchForNonContractingPartitioning() argument
2319 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 …]