-
Notifications
You must be signed in to change notification settings - Fork 5.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[CINN]Add InferSymbolicShape of fpn ops #63947
[CINN]Add InferSymbolicShape of fpn ops #63947
Conversation
Add RoiAlignOpInferSymbolicShape; fix GatherOpInferSymbolicShape when axis input is tensor; fix Expr set_type implementation of derived class;
你的PR提交成功,感谢你对开源项目的贡献! |
对应单测: |
66c4959
to
1f5dfc9
Compare
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
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc
Outdated
Show resolved
Hide resolved
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 for YAML
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc
Outdated
Show resolved
Hide resolved
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc
Outdated
Show resolved
Hide resolved
* Add DistributeFpnProposalsOpInferSymbolicShape; Add RoiAlignOpInferSymbolicShape; fix GatherOpInferSymbolicShape when axis input is tensor; fix Expr set_type implementation of derived class; * fix * change shape_analysis to infer_context * fix
* Add DistributeFpnProposalsOpInferSymbolicShape; Add RoiAlignOpInferSymbolicShape; fix GatherOpInferSymbolicShape when axis input is tensor; fix Expr set_type implementation of derived class; * fix * change shape_analysis to infer_context * fix
PR Category
CINN
PR Types
Bug fixes
Description
Add DistributeFpnProposalsOpInferSymbolicShape;
Add RoiAlignOpInferSymbolicShape;
fix GatherOpInferSymbolicShape when axis input is tensor;