Searched refs:matching_sharding (Results 1 – 1 of 1) sorted by relevance
1881 bool lhs_matching, const HloSharding& matching_sharding, in GetNonContractingPartitionGroupedShardingForMatchedOperand() argument1885 matching_sharding.tile_assignment().dimensions(); in GetNonContractingPartitionGroupedShardingForMatchedOperand()1899 Array<int64> reshaped_matching_tiling = matching_sharding.tile_assignment(); in GetNonContractingPartitionGroupedShardingForMatchedOperand()1903 matching_sharding.ReplicateOnLastTileDim() in GetNonContractingPartitionGroupedShardingForMatchedOperand()2337 const HloSharding& matching_sharding = in LhsIsBestMatchForNonContractingPartitioning() local2375 assume_lhs_match, matching_sharding, output_sharding, in LhsIsBestMatchForNonContractingPartitioning()