diff options
author | 2018-07-26 11:36:30 +0800 | |
---|---|---|
committer | 2018-07-26 11:36:30 +0800 | |
commit | 9bab0c89c4ffeeb780e7a3dc415ab888164b9b00 (patch) | |
tree | ffeebac847f5d1ba5ca551b86533490d337a1193 /tensorflow/python/feature_column | |
parent | 5d92abe1e426b85ef549a8ce811628d708f7d914 (diff) |
fix: No need to convert to tensor when using ResourceVariable in embedding_lookup,
because ResourceVariable support ResourceGather OP.
Diffstat (limited to 'tensorflow/python/feature_column')
-rw-r--r-- | tensorflow/python/feature_column/feature_column_v2.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/python/feature_column/feature_column_v2.py b/tensorflow/python/feature_column/feature_column_v2.py index b4dd23f58d..220a4f7ed6 100644 --- a/tensorflow/python/feature_column/feature_column_v2.py +++ b/tensorflow/python/feature_column/feature_column_v2.py @@ -3283,9 +3283,10 @@ def _safe_embedding_lookup_sparse(embedding_weights, raise ValueError('Missing embedding_weights %s.' % embedding_weights) dtype = sparse_weights.dtype if sparse_weights is not None else None - embedding_weights = [ - ops.convert_to_tensor(w, dtype=dtype) for w in embedding_weights - ] + if not isinstance(embedding_weights[0], resource_variable_ops.ResourceVariable): + embedding_weights = [ + ops.convert_to_tensor(w, dtype=dtype) for w in embedding_weights + ] with ops.name_scope(name, 'embedding_lookup', embedding_weights + [sparse_ids, |