Searched refs:safe_embedding_lookup_sparse_v2 (Results 1 – 2 of 2) sorted by relevance
867 embedding_ops.safe_embedding_lookup_sparse_v2(882 embedding_ops.safe_embedding_lookup_sparse_v2(899 embedding_ops.safe_embedding_lookup_sparse_v2(915 embedding_ops.safe_embedding_lookup_sparse_v2(947 embedding_ops.safe_embedding_lookup_sparse_v2(962 embedding_ops.safe_embedding_lookup_sparse_v2(981 embedding_ops.safe_embedding_lookup_sparse_v2(999 embedding_ops.safe_embedding_lookup_sparse_v2(
627 def safe_embedding_lookup_sparse_v2(embedding_weights, function