Searched refs:FirstShardingDimWithPartitionOfSize (Results 1 – 1 of 1) sorted by relevance
327 int64 FirstShardingDimWithPartitionOfSize(int64 num_partitions, in FirstShardingDimWithPartitionOfSize() function586 FirstShardingDimWithPartitionOfSize(num_partitions, output_sharding); in PartitionBaseCase()2395 const int64 output_sharding_dim = FirstShardingDimWithPartitionOfSize( in LhsIsBestMatchForNonContractingPartitioning()