Method Partitioner

tensorflow/core/kernels/unique_ali_op_util.h:105–115

struct Partitioner {
 public:
  explicit Partitioner(int64 work_size, int32 num_parts) {
    if (work_size <= 0 || num_parts <= 0) { return; }
    num_parts_ = num_parts;
    parts_.reserve(num_parts);
    int64 start = 0;
    for (int32 i = 0; i < num_parts; ++i) {
      int64 end = start + (work_size + i) / num_parts;
      parts_.emplace_back(Range(start, end));
      start = end;
    }
  }

  const Range* GetRange(const int32 id) const {
    if (id < 0 || id >= num_parts_) { return nullptr; }