Method set_data

oneflow/core/framework/tensor.cpp:76–88

  }
}
Maybe<void> Parameter::set_data(const std::shared_ptr<Tensor>& other) {
  if (is_local() && is_eager()) {
    auto rematable_storage = std::dynamic_pointer_cast<vm::RematableTensorStorage>(
        CHECK_JUST(tensor_->eager_blob_object())->tensor_storage());
    bool enable_remat = rematable_storage != nullptr && tensor_->is_local() && tensor_->is_eager();
    if (enable_remat) { rematable_storage->set_eviction_disabled(false); }
    JUST(tensor_->set_data(other));
    if (enable_remat) { rematable_storage->set_eviction_disabled(true); }
  } else {
    JUST(tensor_->set_data(other));
  }
  return Maybe<void>::Ok();
}

std::shared_ptr<Tensor> Parameter::contiguous() const {
  const auto& tensor = std::const_pointer_cast<Tensor>(shared_from_this());