提交 5fb28565 编写于 作者: J jerrywgz

test_develop

上级 e2044c09
......@@ -35,7 +35,7 @@ class BoxClipOp : public framework::OperatorWithKernel {
"The last dimension of InputBox must be 4");
PADDLE_ENFORCE_EQ(im_info_dims.size(), 2,
"The rank of Input(InputBox) in BoxClipOp must be 2");
PADDLE_ENFORCE_EQ(im_info_dims[1], 2,
PADDLE_ENFORCE_EQ(im_info_dims[1], 3,
"The last dimension of ImInfo must be 3");
}
ctx->ShareDim("InputBox", /*->*/ "OutputBox");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册