-
Notifications
You must be signed in to change notification settings - Fork 617
Add Addons namespace to Custom Ops #516
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
Thanks very much Tomer this is a big help. Mind quickly formatting with: |
Done! |
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.
@tomerk Thanks for picking this up! Seems we miss hardshrink
operation here (have to merge master though). Also, would you mind unifying all #ifndef
, #define
and #endif
like TENSORFLOW_ADDONS_FOLDERNAME_KERNELS_OPNAME
. I see some of them not follow the same style. Thanks!
* Port bug fix in TF contrib to addons. Original change at tensorflow/tensorflow@a913689. * Fix lint warning.
…w#464) * check pass through and do the expand_dims() only if needed * add indent to the fixed line * merge return condition to if state
* add hardshrink kernel * make linter happy
* Fix SequenceLoss incompatibility with Keras built-in loops * Remove debugging prints * Change the attribute existence checking to use more pythonic way
* Replace some compat.v1 APIs by their v2 equivalent * Fix lint error
All (the pull request submitter and all commit authors) CLAs are signed, but one or more commits were authored or co-authored by someone other than the pull request submitter. We need to confirm that all authors are ok with their commits being contributed to this project. Please have them confirm that by leaving a comment that contains only Note to project maintainer: There may be cases where the author cannot leave a comment, or the comment is not properly detected as consent. In those cases, you can manually confirm consent of the commit author(s), and set the ℹ️ Googlers: Go here for more info. |
Done, but it looks like the way I rebased onto master messed up the CLA bot. |
No worries we can override CLA when time to merge. |
# Conflicts: # tensorflow_addons/activations/hardshrink.py # tensorflow_addons/custom_ops/activations/cc/kernels/hardshrink_op.cc # tensorflow_addons/custom_ops/activations/cc/kernels/hardshrink_op.h # tensorflow_addons/custom_ops/activations/cc/kernels/hardshrink_op_gpu.cu.cc # tensorflow_addons/custom_ops/activations/cc/ops/hardshrink_op.cc
Nice job, Tomer! |
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.
Thank you, Tomer :-)
@ewilderj Hi, Edd, could you help override cla status? Thanks 😄 |
@WindQAQ @fsx950223 Hi, could you take another look? I'd like to merge it if you both agree :-) |
|
Thanks, all! |
Fixes issue #379
#379
This requires the nightly or the upcoming TF2 RC2, it does not work with the TF2 RC1