-
Notifications
You must be signed in to change notification settings - Fork 1.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
[XPU]. Fixed the mode error in pad3d. #9506
Conversation
Thanks for your contribution! |
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.
see the details
padding[2] = pads[2]; | ||
padding[3] = pads[3]; | ||
padding[4] = pads[0]; | ||
padding[5] = pads[1]; |
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.
pads可能是一个元素吗?
lite/kernels/xpu/pad3d_compute.cc
Outdated
in_dims[4], | ||
in_dims[1], | ||
in_dims[2], | ||
in_dims[3], |
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.
建议使用前面的 n c d h w这些变量作为参数 ---- 理论上不应该存在is_ncdhw为两种值时调用两次api
b30e2ea
to
d2aebe8
Compare
LGTM |
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
修复了xpu上pad3d算子所有模式都使用constant的错误。