未验证 提交 c49e6049 编写于 作者: Z Zhaolong Xing 提交者: GitHub

Merge pull request #16213 from qingqing01/compile_infer_shape

Skip compile infer shape in box_coder_op
...@@ -60,14 +60,15 @@ class BoxCoderOp : public framework::OperatorWithKernel { ...@@ -60,14 +60,15 @@ class BoxCoderOp : public framework::OperatorWithKernel {
} else if (code_type == BoxCodeType::kDecodeCenterSize) { } else if (code_type == BoxCodeType::kDecodeCenterSize) {
PADDLE_ENFORCE_EQ(target_box_dims.size(), 3, PADDLE_ENFORCE_EQ(target_box_dims.size(), 3,
"The rank of Input TargetBox must be 3"); "The rank of Input TargetBox must be 3");
PADDLE_ENFORCE(axis == 0 || axis == 1, "axis must be 0 or 1");
if (ctx->IsRuntime()) {
if (axis == 0) { if (axis == 0) {
PADDLE_ENFORCE_EQ(target_box_dims[1], prior_box_dims[0]); PADDLE_ENFORCE_EQ(target_box_dims[1], prior_box_dims[0]);
} else if (axis == 1) { } else if (axis == 1) {
PADDLE_ENFORCE_EQ(target_box_dims[0], prior_box_dims[0]); PADDLE_ENFORCE_EQ(target_box_dims[0], prior_box_dims[0]);
} else {
PADDLE_THROW("axis must be 0 or 1.");
} }
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");
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册