fix: fix bn infershape bug (#3019)

9590d552 · xiaogang · GitHub · 6989f47f · 9590d552
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

lite/operators/batch_norm_op.cc lite/operators/batch_norm_op.cc +1 -0

未找到文件。
--- a/lite/operators/batch_norm_op.cc
+++ b/lite/operators/batch_norm_op.cc
@@ -68,6 +68,7 @@ bool BatchNormOp::InferShape() const {
    param_.saved_variance->Resize({channel_size});
  }
  param_.y->Resize(x_dims);
+  param_.y->set_lod(param_.x->lod());
  return true;
 }