-
Notifications
You must be signed in to change notification settings - Fork 92
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
Initial implementation of gpu_embedding. #2114
base: develop-upstream
Are you sure you want to change the base?
Conversation
The second part of the changes are in: |
9ae9192
to
7f0c638
Compare
@zoranjovanovic-ns it seems failed at cpu test //tensorflow/tools/api/tests:api_compatibility_test You need to check and reproduce it in your local. |
retest code-check please |
retest cpu-non-pip please |
I am running the tests locally, as soon as it pass I will retest everything. |
3b34e39
to
ee5302b
Compare
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.
LGTM
New gpu mid level API is not intended for large embedding table lookup.
Embedding tables will be replicated across devices rather than sharding across them.