fix: fix op dropout and fc infershape lod bug (#2732)

4430df40 · xiaogang · GitHub · b918067c · 4430df40 · 4430df40
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

lite/operators/dropout_op.cc lite/operators/dropout_op.cc +1 -1

lite/operators/fc_op.cc lite/operators/fc_op.cc +1 -1

未找到文件。
--- a/lite/operators/dropout_op.cc
+++ b/lite/operators/dropout_op.cc
@@ -33,7 +33,7 @@ bool DropoutOp::InferShape() const {
    param_.mask->Resize(x_dims);
  }
  // share LoD
-  // param_.output->set_lod(param_.input->lod());
+  param_.output->set_lod(param_.x->lod());
  return true;
 }

--- a/lite/operators/fc_op.cc
+++ b/lite/operators/fc_op.cc
@@ -61,7 +61,7 @@ bool FcOpLite::InferShape() const {
  param_.output->Resize(lite::DDim(output_dims));
  // share LoD
-  // param_.output->set_lod(param_.input->lod());
+  param_.output->set_lod(param_.input->lod());
  return true;
 }