Skip to content
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

Public function to register custom ops #1193

Merged
merged 15 commits into from
Mar 7, 2020
Merged

Public function to register custom ops #1193

merged 15 commits into from
Mar 7, 2020

Conversation

gabrieldemarmiesse
Copy link
Member

@gabrieldemarmiesse gabrieldemarmiesse commented Mar 1, 2020

Fix #1151

I'll open an issue to track the registering of keras objects with this function. It's there for future backward compatibility.

@seanpmorgan
Copy link
Member

Haven't had time for a full review, but at first glance looks like a good strategy. @guillaumekln could you take a look when time allows and see if this covers the issue from your perspective.

Copy link
Contributor

@guillaumekln guillaumekln left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks good to me, thanks for working on this issue!

tensorflow_addons/register.py Show resolved Hide resolved
tensorflow_addons/register.py Outdated Show resolved Hide resolved
@gabrieldemarmiesse
Copy link
Member Author

Thanks a lot for the review! Let me fix all that.

guillaumekln
guillaumekln previously approved these changes Mar 5, 2020
@guillaumekln
Copy link
Contributor

@seanpmorgan Do you want to take a second look before merging?

@gabrieldemarmiesse
Copy link
Member Author

This pull request will be discussed at the meeting tonight I believe.

@seanpmorgan
Copy link
Member

@gabrieldemarmiesse Conflicts and then LGTM. Per discussion we'll want to work toward registering Keras objects in a similar manner

@gabrieldemarmiesse
Copy link
Member Author

gabrieldemarmiesse commented Mar 7, 2020

Per discussion we'll want to work toward registering Keras objects in a similar manner

Let's do that in another pull request to make the review easier.

Copy link
Member

@seanpmorgan seanpmorgan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM Thanks!

@seanpmorgan seanpmorgan merged commit 1458f7f into tensorflow:master Mar 7, 2020
jrruijli pushed a commit to jrruijli/addons that referenced this pull request Dec 23, 2020
* Added public functions to register everything.

* Removed decorator

* Revert "Removed decorator"

This reverts commit ebea5bd.

* Added some tests.

* Added the two register.

* Removed unused variables.

* Private func.

* Explicit modules.

* FLake8

* Added documentation.

* Remove useless setup method.

* Black/

* Format BUILD.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Public API to force load custom ops
4 participants