experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix the embedding variable name on TPU for v2 shared embedding columns to match the name on the CPU side.

Browse Source 
PiperOrigin-RevId: 254019702
This commit is contained in:
Bruce Fontaine 2019-06-19 10:14:00 -07:00 committed by TensorFlower Gardener
parent 1b0d51abed
commit 6523a23d55
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/tpu/feature_column_v2.py
View File

@ -502,7 +502,7 @@ class _TPUSharedEmbeddingColumnV2(_TPUBaseEmbeddingColumn,
# Note that in Feature Column V2, shared embeddings have no scope.
_record_variable_scope_and_name(
self.get_embedding_var_name(),
'embedding_weights',
self.shared_embedding_column_creator._name,
is_shared_embedding=True)
return tensor