diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-22 10:57:06 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-22 10:57:06 -0700 |
commit | 425e96f3ae4eb338268e3738260f9d79e4bdd893 (patch) | |
tree | 9378c76d616c776f93b177d3273dbb34232f1c78 /tensorflow/python/feature_column | |
parent | 1a8dd7910eedfea6ba1917c8055fcd7fed9b157e (diff) | |
parent | ba5d214a6b5d131b693eff277cc3b56298a4721a (diff) |
Merge pull request #20539 from candyzone:master
PiperOrigin-RevId: 214121495
Diffstat (limited to 'tensorflow/python/feature_column')
-rw-r--r-- | tensorflow/python/feature_column/feature_column_v2.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/python/feature_column/feature_column_v2.py b/tensorflow/python/feature_column/feature_column_v2.py index 28c5c82d2c..57f7af7635 100644 --- a/tensorflow/python/feature_column/feature_column_v2.py +++ b/tensorflow/python/feature_column/feature_column_v2.py @@ -3433,9 +3433,11 @@ 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, |