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
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: