“f6c9f56394838021af5db26d046f3c90606a17fc”上不存在“develop/doc/howto/cluster/multi_cluster/fabric_en.html”
未验证 提交 9d0597ea 编写于 作者: X xiaogang 提交者: GitHub

fix: fix bn infershape bug (#3019)

上级 b87299fe
...@@ -68,6 +68,7 @@ bool BatchNormOp::InferShape() const { ...@@ -68,6 +68,7 @@ bool BatchNormOp::InferShape() const {
param_.saved_variance->Resize({channel_size}); param_.saved_variance->Resize({channel_size});
} }
param_.y->Resize(x_dims); param_.y->Resize(x_dims);
param_.y->set_lod(param_.x->lod());
return true; return true;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册