Searched refs:scattered_embedding_lookup (Results 1 – 2 of 2) sorted by relevance
295 embedding_lookup_result = embedding_ops.scattered_embedding_lookup(307 embedding_lookup_result = embedding_ops.scattered_embedding_lookup(325 embedding_lookup_result = embedding_ops.scattered_embedding_lookup(336 embedding_lookup_result = embedding_ops.scattered_embedding_lookup(
212 def scattered_embedding_lookup(params, function444 embeddings = scattered_embedding_lookup(