Searched refs:other_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
1913 int64 other_contracting_partitions, int64 other_non_contracting_partitions, in GetNonContractingPartitionGroupedShardingForOtherOperand() argument1936 (other_contracting_partitions == group_count && in GetNonContractingPartitionGroupedShardingForOtherOperand()1941 matching_contracting_partitions == other_contracting_partitions; in GetNonContractingPartitionGroupedShardingForOtherOperand()1981 int64 matching_contracting_partitions, int64 other_contracting_partitions, in PartitionDotGroupOnNonContracting() argument2021 other_contracting_partitions, other_non_contracting_partitions, in PartitionDotGroupOnNonContracting()2352 const int64 other_contracting_partitions = in LhsIsBestMatchForNonContractingPartitioning() local2380 other_partitioned.hlo()->shape(), other_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()