From 57d2abb19bdd0452cb935432bbe1427d1eb9d2ce Mon Sep 17 00:00:00 2001 From: "tongxuan.ltx" Date: Mon, 25 Apr 2022 10:11:40 +0800 Subject: [PATCH] [Embedding] Fix embedding_lookup_sparse sub-graph issue when use Variable. --- tensorflow/python/ops/embedding_ops.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/python/ops/embedding_ops.py b/tensorflow/python/ops/embedding_ops.py index a85e75e4c39..1199b9c86da 100644 --- a/tensorflow/python/ops/embedding_ops.py +++ b/tensorflow/python/ops/embedding_ops.py @@ -589,11 +589,11 @@ def embedding_lookup_sparse(params, segment_ids = math_ops.cast(segment_ids, dtypes.int32) ids = sp_ids.values - 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: + ids, idx, counts = array_ops.unique_with_counts(ids) + else: ids, idx = array_ops.unique(ids) counts = None - else: - ids, idx, counts = array_ops.unique_with_counts(ids) uniqued_blocknums = None if blocknums is not None: