Function InitializeVariable

paddle/fluid/framework/variable_helper.cc:28–60

namespace paddle::framework {

void InitializeVariable(Variable *var, proto::VarType::Type var_type) {
  if (var_type == proto::VarType::DENSE_TENSOR) {
    var->GetMutable<DenseTensor>();
  } else if (var_type == proto::VarType::SELECTED_ROWS) {
    var->GetMutable<phi::SelectedRows>();
  } else if (var_type == proto::VarType::FEED_MINIBATCH) {
    var->GetMutable<FeedList>();
  } else if (var_type == proto::VarType::FETCH_LIST) {
    var->GetMutable<FetchList>();
  } else if (var_type == proto::VarType::STEP_SCOPES) {
    var->GetMutable<std::vector<framework::Scope *>>();
  } else if (var_type == proto::VarType::DENSE_TENSOR_ARRAY) {
    var->GetMutable<phi::TensorArray>();
  } else if (var_type == proto::VarType::STRINGS) {
    var->GetMutable<Strings>();
  } else if (var_type == proto::VarType::VOCAB) {
    var->GetMutable<Vocab>();
  } else if (var_type == proto::VarType::PLACE_LIST) {
    var->GetMutable<phi::PlaceList>();
  } else if (var_type == proto::VarType::READER) {
    var->GetMutable<ReaderHolder>();
  } else if (var_type == proto::VarType::RAW) {
    // GetMutable will be called in operator
  } else if (var_type == proto::VarType::SPARSE_COO) {
    var->GetMutable<phi::SparseCooTensor>();
  } else {
    PADDLE_THROW(common::errors::Unavailable(
        "Variable type %d is not in "
        "[DENSE_TENSOR, SELECTED_ROWS, FEED_MINIBATCH, FETCH_LIST, "
        "LOD_RANK_TABLE, PLACE_LIST, READER, RAW].",
        var_type));
  }
}

void CopyVariable(const Variable &src_var, Variable *dst_var) {
  // only support cpu now