-
Notifications
You must be signed in to change notification settings - Fork 738
Arm backend: Add tanh operator #6226
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
Arm backend: Add tanh operator #6226
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/6226
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 8620371 with merge base ca47839 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@pytorchbot label ciflow/trunk |
|
@pytorchbot label 'partner: arm' |
| def tanh(x): | ||
| # Convert quantized input to floating point tanh input space. | ||
| v = dequantize_value(x, in_quantargs) | ||
| # Compute tanh. |
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.
curious why not numpy.tanh or torch.nn.Tanh?
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.
Hi,
Thanks for review! I just kept this consistent with sigmoid.: #4114
I am happy to update this though if needed.
Thanks,
Saoirse
|
@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
Seems like a flaky CI - rerunning it. |
|
Rebase please? CI seems happy now |
Thanks, done! |
|
@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
1 similar comment
|
@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
Adds Tanh operator to the Arm backend including unit tests