Searched defs:embedding_weights (Results 1 – 2 of 2) sorted by relevance
685 def safe_embedding_lookup_sparse_v2(embedding_weights, argument787 def safe_embedding_lookup_sparse(embedding_weights, argument955 def embedding_lookup_ragged(embedding_weights, argument
2981 embedding_weights): argument3193 def embedding_weights(self): member in SharedEmbeddingColumnCreator