Searched refs:output_batch_partitions (Results 1 – 1 of 1) sorted by relevance
475 int64 output_batch_partitions, int64 lhs_contracting_partitions, in PartitionBaseCase() argument540 output_batch_partitions == num_partitions && in PartitionBaseCase()557 output_batch_partitions == num_partitions && in PartitionBaseCase()1550 if (output_batch_partitions == num_partitions) { in PartitionBaseCase()2079 int64 output_batch_partitions, int64 output_lhs_non_contracting_partitions, in PartitionDotGroupOnContracting() argument2196 output_batch_partitions == group_count) { in PartitionDotGroupOnContracting()2476 const int64 output_batch_partitions = in PartitionDot() local2506 output_batch_partitions > 1 || in PartitionDot()2590 lhs_batch_partitions, rhs_batch_partitions, output_batch_partitions, in PartitionDot()2604 if ((lhs_batch_partitions == output_batch_partitions || in PartitionDot()[all …]