-
Notifications
You must be signed in to change notification settings - Fork 825
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 hard sigmoid #4043
Merged
Merged
Add hard sigmoid #4043
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Ldpe2G
reviewed
Dec 30, 2020
Ldpe2G
reviewed
Dec 30, 2020
Ldpe2G
reviewed
Dec 31, 2020
Ldpe2G
approved these changes
Jan 4, 2021
liujuncheng
pushed a commit
that referenced
this pull request
Jun 3, 2021
* add cpu version * add hardsigmoid kernel * add test case * fix Former-commit-id: 7d4ffee
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
增加 hardsigmoid 算子
暂时没做inplace
Pytorch实现有误区, 边界值依旧有梯度,做不了inplace操作
Paddle的边界值则没有梯度
概述
实现hardsigmiod算子,对标Pytorch的 torch.nn.Hardsigmoid

功能 CheckList
注意 : 功能复选框均为可选项,若未选择,说明理由即可。例如:该 Op 由 Python 接口拼接而成,因此无
SetBatchAxisInferFn
Op 注册;再比如:该 Op 无输入,因此无SetInputArgModifyFn
。模板中自带的复选框可留空,但是不能删除。可根据实际情况增加复选框选项。
Op
Kernel
注册浮点类型, 仅仅针对前向注册 int整数类型
CPU in:float32
CPU in:float64
GPU in:float32
GPU in:float64
GPU in:float16
Python Wrapper
测试
GPU 有效带宽
理论带宽:
实际带宽:
折合成百分比为69.64%
PR Checklist
bug, enhancement, purge, feature, documentation
)op, system, eager, build, xla, python, ci, test, tooling, onnx
)