Searched refs:rhs_batch_partitions (Results 1 – 1 of 1) sorted by relevance
387 int64 rhs_batch_partitions, int64 lhs_contracting_partitions, in GetWindowedEinsumConfiguration() argument411 if (rhs_batch_partitions == num_partitions) { in GetWindowedEinsumConfiguration()474 int64 lhs_batch_partitions, int64 rhs_batch_partitions, in PartitionBaseCase() argument521 if (lhs_batch_partitions == rhs_batch_partitions && in PartitionBaseCase()522 rhs_batch_partitions == num_partitions && in PartitionBaseCase()556 if (rhs_batch_partitions == num_partitions && in PartitionBaseCase()1471 rhs_non_contracting_partitions, rhs_batch_partitions, in PartitionBaseCase()2296 int64 rhs_batch_partitions, bool may_group_on_lhs_non_contracting, in LhsIsBestMatchForNonContractingPartitioning() argument2362 assume_lhs_match ? lhs_batch_partitions : rhs_batch_partitions; in LhsIsBestMatchForNonContractingPartitioning()2364 assume_lhs_match ? rhs_batch_partitions : lhs_batch_partitions; in LhsIsBestMatchForNonContractingPartitioning()[all …]