提交 921e7c1e 编写于 作者: D DannyIsFunny

test=develop

上级 54cbdcb6
...@@ -318,7 +318,7 @@ const lite::Tensor *Predictor::GetTensor(const std::string &name) const { ...@@ -318,7 +318,7 @@ const lite::Tensor *Predictor::GetTensor(const std::string &name) const {
return &var->Get<lite::Tensor>(); return &var->Get<lite::Tensor>();
} }
lite::Tensor *Predictor::GetMutableTensor(const std::string &name) const { lite::Tensor *Predictor::GetMutableTensor(const std::string &name) {
auto *var = exec_scope_->FindVar(name); auto *var = exec_scope_->FindVar(name);
CHECK(var) << "no variable named with " << name << " in exec_scope"; CHECK(var) << "no variable named with " << name << " in exec_scope";
return var->GetMutable<lite::Tensor>(); return var->GetMutable<lite::Tensor>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册