Lines Matching refs:ids
32 def bow_encoder(ids,
44 ids: `[batch_size, doc_length]` `Tensor` or `SparseTensor` of type
45 `int32` or `int64` with symbol ids.
48 sparse_lookup: `bool`, if `True`, converts ids to a `SparseTensor`
71 scope, 'bow_encoder', [ids], reuse=reuse):
77 if isinstance(ids, sparse_tensor.SparseTensor):
78 sparse_ids = ids
80 sparse_ids = sparse_ops.dense_to_sparse_tensor(ids)
84 if isinstance(ids, sparse_tensor.SparseTensor):
85 raise TypeError('ids are expected to be dense Tensor, got: %s', ids)
87 embedding_ops.embedding_lookup(embeddings, ids),
91 def embed_sequence(ids,
105 ids: `[batch_size, doc_length]` `Tensor` of type `int32` or `int64`
106 with symbol ids.
132 scope, 'EmbedSequence', [ids], reuse=reuse):
141 return contrib_embedding_ops.embedding_lookup_unique(embeddings, ids)
142 return embedding_ops.embedding_lookup(embeddings, ids)
