Searched refs:_embedding_lookup_for_sparse_tensor (Results 1 – 2 of 2) sorted by relevance
312 _embedding_lookup_for_sparse_tensor(inp, weight, table, feature))322 def _embedding_lookup_for_sparse_tensor( function
295 self._embedding_lookup_for_sparse_tensor(inp, weight, table,306 def _embedding_lookup_for_sparse_tensor( member in TPUEmbeddingV0