-
Notifications
You must be signed in to change notification settings - Fork 5.7k
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
【Infer Symbolic Shape No.11】【BUAA】 Add box_coder op #67864
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
phi::errors::InvalidArgument( | ||
"The dimension of Input(PriorBoxVar) should be equal to " | ||
"the dimension of Input(PriorBox) in BoxCoder operator " | ||
"when the rank is 2.")); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里需要写个循环,逐元素添加equalcstr
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已修改。
infer_context->SetShapeOrDataForValue( | ||
op->result(0), | ||
symbol::ShapeOrDataDimExprs{symbol::TensorShapeOrDataDimExprs( | ||
{target_box_shape[0], prior_box_shape[0], 4})}); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
顺便改成symbol::DimExpr(4)吧
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已修改。
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
const std::string &code_type = | ||
op->attribute<pir::StrAttribute>("code_type").AsString(); | ||
int axis = op->attribute<pir::Int32Attribute>("axis").data(); | ||
const std::vector<float> &variance = |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
PR Category
CINN
PR Types
Improvements
Description
添加 box_coder 算子符号推导接口。