From 9590d552aa9350a105af90a1aba1dc3a41054e24 Mon Sep 17 00:00:00 2001
From: xiaogang <txg4794@163.com>
Date: Wed, 4 Mar 2020 15:23:00 +0800
Subject: [PATCH] fix: fix bn infershape bug (#3019)

---
 lite/operators/batch_norm_op.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lite/operators/batch_norm_op.cc b/lite/operators/batch_norm_op.cc
index 76c257c6d3..eca7fa6001 100644
--- 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;
 }
 
-- 
GitLab