提交 0d915078 编写于 作者: J jerrywgz

fix share lod, test=develop

上级 ab9d6a4f
...@@ -77,10 +77,14 @@ class BoxCoderOp : public framework::OperatorWithKernel { ...@@ -77,10 +77,14 @@ class BoxCoderOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(target_box_dims[2], prior_box_dims[1]); PADDLE_ENFORCE_EQ(target_box_dims[2], prior_box_dims[1]);
ctx->ShareDim("TargetBox", /*->*/ "OutputBox"); ctx->ShareDim("TargetBox", /*->*/ "OutputBox");
} }
}
if (code_type == BoxCodeType::kDecodeCenterSize && axis == 1) {
ctx->ShareLoD("PriorBox", /*->*/ "OutputBox");
} else {
ctx->ShareLoD("TargetBox", /*->*/ "OutputBox"); ctx->ShareLoD("TargetBox", /*->*/ "OutputBox");
} }
}
}
}; };
class BoxCoderOpMaker : public framework::OpProtoAndCheckerMaker { class BoxCoderOpMaker : public framework::OpProtoAndCheckerMaker {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册