load op add seletedRows

7 years ago · ee64f577d4
parent ca27f78e29
commit ee64f577d4
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/load_op.cc
+++ b/paddle/fluid/operators/load_op.cc
@ -71,7 +71,7 @@ class LoadOp : public framework::OperatorBase {

      auto *tensor = var->GetMutable<framework::LoDTensor>();

-    DeserializeFromStream(fin, tensor, *dev_ctx);
+    DeserializeFromStream(fin, tensor, dev_ctx);

    auto load_as_fp16 = Attr<bool>("load_as_fp16");
    auto in_dtype = framework::ToDataType(tensor->type());