Skip to content

Commit

Permalink
refactor sok dynamic variable
Browse files Browse the repository at this point in the history
  • Loading branch information
chengmengli06 committed Jan 4, 2024
2 parents 2d6952b + fddb73a commit 8530ca8
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion easy_rec/python/compat/feature_column/feature_column.py
Original file line number Diff line number Diff line change
Expand Up @@ -327,7 +327,7 @@ def embedding_parallel_lookup(embedding,
recv_embeddings = array_ops.gather(embedding, all_uniq_ids)
embeddings = math_ops.sparse_segment_sum(
recv_embeddings, uniq_idx, segment_ids, name='sparse_segment_sum')
# all_embed = array_ops.gather(recv_embeddings, uniq_idx)

embed_dim = embedding.get_shape()[-1]
output_tensor = array_ops.reshape(embeddings, [N, -1, embed_dim])

Expand Down Expand Up @@ -486,6 +486,7 @@ def _get_logits_embedding_parallel(): # pylint: disable=missing-docstring
lookup_output_ids.append(output_id)
lookup_cols.append(column)
lookup_combiners.append(column.combiner)

# SparseTensor RaggedTensor
# features are not gathered into one, may have
# performance issues
Expand Down

0 comments on commit 8530ca8

Please sign in to comment.