Searched refs:should_partition_contracting_dim (Results 1 – 1 of 1) sorted by relevance
1584 const auto should_partition_contracting_dim = [&](int64 operand_idx) { in PartitionBaseCase() local1607 if (output_sharding.IsReplicated() && (should_partition_contracting_dim(0) || in PartitionBaseCase()1608 should_partition_contracting_dim(1))) { in PartitionBaseCase()1611 if (should_partition_contracting_dim(0)) { in PartitionBaseCase()