Add a helper method to get the embedding ids NestedMap for lookup. Add a check in EmbLookup[Sparse] to make sure the input keys are valid.
Add a helper method to get the embedding ids NestedMap for lookup.
Add a check in EmbLookup[Sparse] to make sure the input keys are valid.

PiperOrigin-RevId: 452650728

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo