-
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
[Eager] Supported Eager Dygraph Forward & AutoGrad functions #37323
[Eager] Supported Eager Dygraph Forward & AutoGrad functions #37323
Conversation
✅ This PR's description meets the template requirements! |
Thanks for your contribution! |
7caf12d
to
a74d2a3
Compare
a74d2a3
to
6eee285
Compare
6eee285
to
4e208a0
Compare
… dygraph_second_pr1
… dygraph_second_pr1
c16e972
to
6bd9e9f
Compare
6bd9e9f
to
a08394c
Compare
} | ||
} | ||
|
||
void ScaleAPI(const egr::EagerTensor& x, float scale, float bias, |
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.
eager API 后续这块计划是怎样的?暂时均以手写封装推进吗?
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.
还是说在pten api解决目前的性能问题之前,均会以这种方式独立编写?
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.
没有,这里主要是因为之前所有测试都是以Scale为基础写的,但是Pten暂时没迁Scale
如果后面Scale迁移了我会替换成Pten版的
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
New features
PR changes
Others
Describe
This patch includes: