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,
framework::Variable *var) {
void SaveLodTensor(const std::string &filename, const platform::Place &place,
framework::Variable *var) const {
auto &tensor = var->Get<framework::LoDTensor>();
// get device context from pool
@ -128,8 +128,9 @@ class SaveOp : public framework::OperatorBase {
fout.close();
}
SaveSelectedRows(const std::string &filename, const platform::Place &place,
framework::Variable *var) {
void SaveSelectedRows(const std::string &filename,
const platform::Place &place,
framework::Variable *var) const {
auto &selectedRows = var->Get<framework::SelectedRows>();
// get device context from pool

Loading…
Cancel
Save