Method LocatePos

tensorflow/core/kernels/unique_ali_op_util.h:122–130

  }

  bool LocatePos(const int64 pos, int32* task_id) const {
    for (int32 i = 0; i < num_parts_; ++i) {
      if (pos >= parts_[i].Start() && pos < parts_[i].End()) {
        *task_id = i;
        return true;
      }
    }
    return false;
  }

 private:
  std::vector<Range> parts_;