-
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
support the 'generate_proposals' for generating static graph ops #52940
Changes from 3 commits
3c63a22
4334702
4061de6
edf7973
b0ce785
46eade5
2099fc6
4d35bf9
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
This file was deleted.
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -780,6 +780,16 @@ | |
func : gelu | ||
backward : gelu_grad | ||
|
||
- op : generate_proposals | ||
args : (Tensor scores, Tensor bbox_deltas, Tensor im_shape, Tensor anchors, Tensor variances, int pre_nms_top_n, int post_nms_top_n, float nms_thresh, float min_size, float eta, bool pixel_offset=true) | ||
output : Tensor(rpn_rois), Tensor(rpn_roi_probs), Tensor(rpn_rois_num) | ||
infer_meta : | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 增加 |
||
func : GenerateProposalsV2InferMeta | ||
kernel : | ||
func : generate_proposals | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 增加 |
||
optional : rpn_rois_num | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 参考其他ops的 |
||
data_type : anchors | ||
|
||
- op : grid_sample | ||
args : (Tensor x, Tensor grid, str mode = "bilinear", str padding_mode = "zeros", bool align_corners = true) | ||
output : Tensor(out) | ||
|
This file was deleted.
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.
可以通过
ci-py3
的错误信息同时结合xxx_op.cc文件,这里有部分attrs的name名字也需要映射。仔细对比xxx_op.cc
文件和yaml中的名字进行修改,如"pre_nms_top_n",请全面对比名字。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.
好的👌