Searched refs:embedding_lookup_ragged (Results 1 – 3 of 3) sorted by relevance
317 return embedding_lookup_ragged(params, ids,943 def embedding_lookup_ragged(embedding_weights, function
1755 embedded_ragged = nn.embedding_lookup_ragged(weights, ragged_ids)1771 embedded_ragged = nn.embedding_lookup_ragged(weights, ragged_ids)1785 nn.embedding_lookup_ragged(None, ragged_ids)1792 nn.embedding_lookup_ragged([], ragged_ids)1800 nn.embedding_lookup_ragged(weights, ragged_ids)
1495 ragged_result = embedding_ops.embedding_lookup_ragged(table, ragged)