Searched defs:lhs_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
387 int64 rhs_batch_partitions, int64 lhs_contracting_partitions, in GetWindowedEinsumConfiguration()475 int64 output_batch_partitions, int64 lhs_contracting_partitions, in PartitionBaseCase()1651 int64 num_partitions, int64 lhs_contracting_partitions, in PartitionDotGroupOnBatch()2292 int64 rhs_non_contracting_partitions, int64 lhs_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()2478 const int64 lhs_contracting_partitions = in PartitionDot() local