提交 98c30c7c 编写于 作者: T tangwei12

bug fix

上级 fe76244f
无相关合并请求
...@@ -96,8 +96,8 @@ class SaveOp : public framework::OperatorBase { ...@@ -96,8 +96,8 @@ class SaveOp : public framework::OperatorBase {
} }
} }
SaveLodTensor(const std::string &filename, const platform::Place &place, void SaveLodTensor(const std::string &filename, const platform::Place &place,
framework::Variable *var) { framework::Variable *var) const {
auto &tensor = var->Get<framework::LoDTensor>(); auto &tensor = var->Get<framework::LoDTensor>();
// get device context from pool // get device context from pool
...@@ -128,8 +128,9 @@ class SaveOp : public framework::OperatorBase { ...@@ -128,8 +128,9 @@ class SaveOp : public framework::OperatorBase {
fout.close(); fout.close();
} }
SaveSelectedRows(const std::string &filename, const platform::Place &place, void SaveSelectedRows(const std::string &filename,
framework::Variable *var) { const platform::Place &place,
framework::Variable *var) const {
auto &selectedRows = var->Get<framework::SelectedRows>(); auto &selectedRows = var->Get<framework::SelectedRows>();
// get device context from pool // get device context from pool
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册
反馈
建议
客服 返回
顶部