port
tangwei12 7 years ago
parent fe76244f0e
commit 98c30c7cbe

@ -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 {
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

Loading…
Cancel
Save