diff --git a/src/framework/operator.cpp b/src/framework/operator.cpp index 2b9e1d04514f2da304b3a35af81d21442817f93d..611b134eaab7376dfba9ef3294f8fa14bf13e0da 100644 --- a/src/framework/operator.cpp +++ b/src/framework/operator.cpp @@ -64,8 +64,7 @@ void OperatorBase::Run() { for (const auto key : input_keys) { auto var_vec_in = inputs_.at(key); for (int i = 0; i < var_vec_in.size(); ++i) { - DLOG << var_vec_in[i]; - auto vari = this->scope_->FindVar("input"); + auto vari = this->scope_->FindVar(var_vec_in[i]); if (vari->IsInitialized()) { const Tensor *tensor = vari->template Get(); if (tensor) DLOG << type_ << " input- " << key << "=" << *tensor;