load op add seletedRows

ee64f577 · tangwei12 · ca27f78e · ee64f577
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/load_op.cc paddle/fluid/operators/load_op.cc +1 -1

未找到文件。
--- 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());