-
Notifications
You must be signed in to change notification settings - Fork 661
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
add broadcast pow #6013
add broadcast pow #6013
Conversation
@@ -101,6 +101,18 @@ | |||
signature: "Tensor (Tensor x, Scalar alpha) => ScalarPow" | |||
bind_python: False | |||
|
|||
- name: "broadcast_pow" |
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.
这里是不是可以绑一下python.,其它的broadcast op都绑了
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.
这里直接仿照add接口重新写一下pow的接口吧
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.
这个地方没有绑定python是因为迟哥之前把pow这个接口在oneflow/api/python/functional/python_functions.cpp导出了,在上述文件中导出的接口调用了scalar_pow和element-wise pow,所以我就没有导出而是直接替换了element-wise pow
No description provided.