Method Pad

tensorflow/core/framework/hash_table/tensible_variable.cc:117–147

}

void TensibleVariable::Pad(
    int64 size, const std::function<void(Status)>& done) {
  if (size == size_) {
    done(Status::OK());
    return;
  }
  if (size_ - size > segment_size_ || size_ < size) {
    done(errors::FailedPrecondition(
          "TensibleVariable Pad size should less than 1 segment ",
          std::to_string(size_), " ", std::to_string(size)));
    return;
  }
  generator_->GetNextTensor([this, size, done](Status st, const Tensor& tensor) {
    if (!st.ok()) {
      done(st);
      return;
    }
    if (size_ - size > segment_size_ || size_ < size) {
      done(errors::FailedPrecondition(
          "TensibleVariable Pad size should less than 1 segment ",
          std::to_string(size_), " ", std::to_string(size)));
      return;
    }
    int64 pad = size_ - size;
    memcpy(const_cast<char*>(
        tensors_.back().tensor_data().data()) +
          (segment_size_ - pad) * slice_size_,
        tensor.tensor_data().data(), pad * slice_size_);
    done(Status::OK());
  });
}

void TensibleVariable::Clear() {
  mutex_lock lock(structure_update_mu_);