Searched refs:output_lhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
384 int64 num_partitions, int64 output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration() argument394 if (output_lhs_non_contracting_partitions == num_partitions && in GetWindowedEinsumConfiguration()448 if (output_lhs_non_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration()478 int64 output_lhs_non_contracting_partitions, in PartitionBaseCase() argument1469 num_partitions, output_lhs_non_contracting_partitions, in PartitionBaseCase()1530 output_lhs_non_contracting_partitions == num_partitions && in PartitionBaseCase()1561 if (output_lhs_non_contracting_partitions == num_partitions) { in PartitionBaseCase()2079 int64 output_batch_partitions, int64 output_lhs_non_contracting_partitions, in PartitionDotGroupOnContracting() argument2194 } else if (output_lhs_non_contracting_partitions == group_count || in PartitionDotGroupOnContracting()2197 if (output_lhs_non_contracting_partitions == group_count) { in PartitionDotGroupOnContracting()[all …]