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

fix: tools op add lod infer (#3380)

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