未验证 提交 40a31442 编写于 作者: X xiaogang 提交者: GitHub

fix: tools op add lod infer (#3380)

上级 0e354afc
...@@ -26,6 +26,7 @@ bool CalibOpLite::CheckShape() const { ...@@ -26,6 +26,7 @@ bool CalibOpLite::CheckShape() const {
} }
bool CalibOpLite::InferShapeImpl() const { bool CalibOpLite::InferShapeImpl() const {
param_.output->Resize(param_.input->dims()); param_.output->Resize(param_.input->dims());
param_.output->set_lod(param_.input->lod());
return true; return true;
} }
......
...@@ -26,6 +26,7 @@ bool IoCopyOp::CheckShape() const { ...@@ -26,6 +26,7 @@ bool IoCopyOp::CheckShape() const {
} }
bool IoCopyOp::InferShapeImpl() const { bool IoCopyOp::InferShapeImpl() const {
param_.y->Resize(param_.x->dims()); param_.y->Resize(param_.x->dims());
param_.y->set_lod(param_.x->lod());
return true; return true;
} }
bool IoCopyOp::Run() { return OpLite::Run(); } bool IoCopyOp::Run() { return OpLite::Run(); }
......
...@@ -26,6 +26,7 @@ bool LayoutOp::CheckShape() const { ...@@ -26,6 +26,7 @@ bool LayoutOp::CheckShape() const {
} }
bool LayoutOp::InferShapeImpl() const { bool LayoutOp::InferShapeImpl() const {
param_.y->Resize(param_.x->dims()); param_.y->Resize(param_.x->dims());
param_.y->set_lod(param_.x->lod());
return true; return true;
} }
bool LayoutOp::Run() { return OpLite::Run(); } bool LayoutOp::Run() { return OpLite::Run(); }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册