Searched refs:output_rhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
385 int64 output_rhs_non_contracting_partitions, in GetWindowedEinsumConfiguration() argument419 if (output_rhs_non_contracting_partitions == num_partitions && in GetWindowedEinsumConfiguration()455 if (output_rhs_non_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration()479 int64 output_rhs_non_contracting_partitions, in PartitionBaseCase() argument1470 output_rhs_non_contracting_partitions, rhs_contracting_partitions, in PartitionBaseCase()1540 output_rhs_non_contracting_partitions == num_partitions && in PartitionBaseCase()1571 if (output_rhs_non_contracting_partitions == num_partitions) { in PartitionBaseCase()2080 int64 output_rhs_non_contracting_partitions, const Shape& output_base_shape, in PartitionDotGroupOnContracting() argument2195 output_rhs_non_contracting_partitions == group_count || in PartitionDotGroupOnContracting()2201 } else if (output_rhs_non_contracting_partitions == group_count) { in PartitionDotGroupOnContracting()[all …]