提交 636fbf0e 编写于 作者: W WangLiu 提交者: GitHub

Merge pull request #317 from cocodark/develop

Init tensor memory in executor_for_test
......@@ -23,6 +23,7 @@ vector<string> OperatorBase<Dtype>::GetOutKeys() const {
auto it = op_input_output_key.find(type_);
if (it == op_input_output_key.end()) {
DLOG << type_ << " has no outputs";
return {};
}
return it->second.second;
}
......
......@@ -73,6 +73,7 @@ class Executor4Test : public Executor<DeviceType> {
}
}
}
this->InitMemory();
}
template <typename T = LoDTensor>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册