tensorflow / recommenders-addons

Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.
Apache License 2.0
587 stars 132 forks source link

[feature]support ragged tensor in all2all #428

Closed jq closed 2 months ago

jq commented 3 months ago

Description

support ragged tensor in all2all HvdAllToAllEmbedding lookup
refactor embedding variable, introduce embedding_weights interface so that dynamic embedding variable, shadow variable, hvd variable can be easily supported in the lookup and safe_embedding_lookup_sparse

Brief Description of the PR:

Fixes # (issue)

Type of change

Checklist:

How Has This Been Tested?

If you're adding a bugfix or new feature please describe the tests that you ran to verify your changes: