Searched refs:_embedding_lookup_and_transform (Results 1 – 3 of 3) sorted by relevance
96 …sampled_logits = embedding_ops._embedding_lookup_and_transform( # pylint: disable=protected-access
84 def _embedding_lookup_and_transform(params, function310 return _embedding_lookup_and_transform(
627 simple = embedding_ops._embedding_lookup_and_transform(636 sharded = embedding_ops._embedding_lookup_and_transform(