Searched refs:embedding_lookup_unique (Results 1 – 3 of 3) sorted by relevance
140 return contrib_embedding_ops.embedding_lookup_unique(embeddings, ids)
375 embedded_tf = embedding_ops.embedding_lookup_unique(embeds, idx).eval()388 embedded_tf = embedding_ops.embedding_lookup_unique(embeds, idx).eval()389 embedded_tf_lst = embedding_ops.embedding_lookup_unique([embeds],391 embedded_tf2d = embedding_ops.embedding_lookup_unique(embeds,
462 def embedding_lookup_unique(params, ids, partition_strategy="mod", name=None): function