-
Notifications
You must be signed in to change notification settings - Fork 10.6k
[TF-812] Defines remaining derivatives for tgmath functions. #27953
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
Conversation
The following math functions are now differentiable: * `remainder` * `fmod` * `ceil` * `floor` * `round` * `trunc` Resolves [TF-812](https://bugs.swift.org/projects/TF/issues/TF-812)
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.
Rather than using @differentiable(..., vjp: ...) on the original tgmath functions, could you please use @differentiating on the derivative functions?
Sure, will do that. Commit coming soon. |
|
hello @dan-zheng ... I tried to define the derivates using the |
Thanks for sharing! This issue looks like TF-429: the differentiation transform does not yet support I wouldn't expected |
|
@swift-ci Please test tensorflow |
|
This PR, as it is, compiles fine in my machine. Let’s see what’s the result of the CI. |
|
hi @dan-zheng .. sure, I can do that. I'll start a new PR. |
|
I've started a new PR #28108 |
|
Let's close this in favor of #28108, since it uses |
The following math functions are now differentiable:
remainderfmodceilfloorroundtruncResolves TF-812