Searched defs:embedding_weights (Results 1 – 3 of 3) sorted by relevance
684 def safe_embedding_lookup_sparse_v2(embedding_weights, argument786 def safe_embedding_lookup_sparse(embedding_weights, argument954 def embedding_lookup_ragged(embedding_weights, argument
32 def get_pt_result(embedding_weights, mask, indices_type): argument36 def get_reference_result(embedding_weights, mask, indices_type): argument
2977 embedding_weights): argument3190 def embedding_weights(self): member in SharedEmbeddingColumnCreator