fix: fix bn infershape bug (#3019)

9d0597ea · xiaogang · GitHub · b87299fe · 9d0597ea
隐藏空白更改
内联并排

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;
 }