Searched defs:lhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
388 int64 lhs_non_contracting_partitions, int64 lhs_batch_partitions, in GetWindowedEinsumConfiguration()476 int64 rhs_contracting_partitions, int64 lhs_non_contracting_partitions, in PartitionBaseCase()1652 int64 rhs_contracting_partitions, int64 lhs_non_contracting_partitions, in PartitionDotGroupOnBatch()2291 int64 num_partitions, int64 lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()2482 const int64 lhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local