-
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
[phi] transfer unsqueeze and squeeze to phi #40596
[phi] transfer unsqueeze and squeeze to phi #40596
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.
这个benchmark有补充了吗
benchmark 自带了。 unsqueeze和squeeze都有的 |
paddle/phi/infermeta/unary.h
Outdated
@@ -34,6 +34,11 @@ class MetaConfig; | |||
// | |||
// The InferMeta Functions in this file are arranged in alphabetic order. | |||
|
|||
void UnsqueezeInferMeta(const MetaTensor& x, |
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.
现在这个infermeta有需要按照字典序写的要求,改个位置吧
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 op benchmark
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 PADDLE_ENFORCE
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.
这两个kernel是不是都是设备无关的实现,按理说应该像reshape一样放到kernels根目录下?
好的,下一个PR修改放置位置。 |
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
PR types
Others
PR changes
Others
Describe
[phi] transfer unsqueeze and squeeze to phi