We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c51a61f commit c10129cCopy full SHA for c10129c
tensorflow/python/ops/embedding_ops.py
@@ -589,11 +589,11 @@ def embedding_lookup_sparse(params,
589
segment_ids = math_ops.cast(segment_ids, dtypes.int32)
590
591
ids = sp_ids.values
592
- if isinstance(params[0], kv_variable_ops.EmbeddingVariable) and params[0]._filter_freq == 0:
+ if isinstance(params[0], kv_variable_ops.EmbeddingVariable) and params[0]._filter_freq > 0:
593
+ ids, idx, counts = array_ops.unique_with_counts(ids)
594
+ else:
595
ids, idx = array_ops.unique(ids)
596
counts = None
- else:
- ids, idx, counts = array_ops.unique_with_counts(ids)
597
598
uniqued_blocknums = None
599
if blocknums is not None:
0 commit comments