| 68 | } |
| 69 | |
| 70 | void ShareParamsIntoScope(const std::vector<std::string> ¶m_names, |
| 71 | const std::shared_ptr<VariableMap> ¶ms_dict, |
| 72 | framework::Scope *scope) { |
| 73 | for (auto name : param_names) { |
| 74 | PADDLE_ENFORCE_EQ(params_dict->count(name), |
| 75 | 1, |
| 76 | common::errors::InvalidArgument( |
| 77 | "Parameter named %s is not existed in params_dict. " |
| 78 | "Please check that your model was saved correctly", |
| 79 | name)); |
| 80 | |
| 81 | auto ¶m = params_dict->find(name)->second; |
| 82 | auto &dense_tensor = param->Get<DenseTensor>(); |
| 83 | auto *var = scope->Var(name); |
| 84 | auto *dst_tensor = var->GetMutable<DenseTensor>(); |
| 85 | *dst_tensor = dense_tensor; |
| 86 | } |
| 87 | } |
| 88 | |
| 89 | void RemoveFeedFetch(framework::ProgramDesc *program_desc) { |
| 90 | for (size_t i = 0; i < program_desc->Size(); ++i) { |
no test coverage detected