Searched refs:lhs_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
387 int64 rhs_batch_partitions, int64 lhs_contracting_partitions, in GetWindowedEinsumConfiguration() argument422 if (lhs_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration()444 if (lhs_contracting_partitions == rhs_contracting_partitions && in GetWindowedEinsumConfiguration()445 lhs_contracting_partitions == num_partitions && in GetWindowedEinsumConfiguration()475 int64 output_batch_partitions, int64 lhs_contracting_partitions, in PartitionBaseCase() argument1472 lhs_contracting_partitions, lhs_non_contracting_partitions, in PartitionBaseCase()1495 if (lhs_contracting_partitions == rhs_contracting_partitions && in PartitionBaseCase()1496 lhs_contracting_partitions == num_partitions) { in PartitionBaseCase()1597 return (lhs_contracting_partitions == num_partitions) && in PartitionBaseCase()1651 int64 num_partitions, int64 lhs_contracting_partitions, in PartitionDotGroupOnBatch() argument[all …]