Searched refs:target_partition_count (Results 1 – 2 of 2) sorted by relevance
21 std::vector<int64> ShapePartitionAssigner::Run(int64 target_partition_count) { in Run() argument33 if (outer_dim_size >= target_partition_count) { in Run()39 target_partition_count = std::min(outer_dim_size, target_partition_count); in Run()48 static_cast<double>(target_partition_count), 1.0 / outer_dims.size()); in Run()63 target_partition_count) { in Run()75 target_partition_count / other_dims_partition_count - in Run()
70 std::vector<int64> Run(int64 target_partition_count);