-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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 static graph code-gen for deformableConvV1 #53569
Changes from 2 commits
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 |
---|---|---|
|
@@ -17,6 +17,16 @@ | |
func : deformable_conv_grad | ||
data_type : x | ||
|
||
- backward_op : deformable_conv_v1_grad | ||
forward : deformable_conv_v1 (Tensor x, Tensor offset, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, int[] dilations={1, 1}, int deformable_groups=1, int groups=1, int im2col_step=64) -> Tensor(out) | ||
args : (Tensor x, Tensor offset, Tensor filter,Tensor out_grad, int[] strides, int[] paddings, int[] dilations, int deformable_groups, int groups, int im2col_step) | ||
output : Tensor(x_grad), Tensor(offset_grad), Tensor(filter_grad) | ||
infer_meta : | ||
func : DeformableConvGradInferMeta | ||
kernel : | ||
func : deformable_conv_v1_grad | ||
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.
|
||
data_type : x | ||
|
||
- backward_op : embedding_grad | ||
forward : embedding (Tensor x, Tensor weight, int64_t padding_idx=-1) -> Tensor(out) | ||
args : (Tensor x, Tensor weight, Tensor out_grad, int64_t padding_idx=-1) | ||
|
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -88,6 +88,16 @@ | |
data_type : x | ||
backward : deformable_conv_grad | ||
|
||
- op : deformable_conv_v1 | ||
args : (Tensor x, Tensor offset, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, int[] dilations={1, 1}, int deformable_groups=1, int groups=1, int im2col_step=64) | ||
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.yaml中和backward中 |
||
output : Tensor(out) | ||
infer_meta : | ||
func : DeformableConvInferMeta | ||
kernel : | ||
func : deformable_conv_v1 | ||
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.
|
||
data_type : x | ||
backward : deformable_conv_v1_grad | ||
|
||
- op : embedding | ||
args : (Tensor x, Tensor weight, int64_t padding_idx=-1) | ||
output : Tensor | ||
|
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.
deformable_conv_v1
->deformable_conv(deformable_conv_v1)
deformable_conv_v1_grad
->deformable_conv_grad(deformable_conv_v1_grad)