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

Merge pull request #31238 from guillaumekln:patch-3

Browse Source 
PiperOrigin-RevId: 261231617
This commit is contained in:
TensorFlower Gardener 2019-08-01 19:44:15 -07:00
commit 216400a8aa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/ops/lookup_ops.py
View File

@ -1025,7 +1025,7 @@ class IdTableWithHashBuckets(LookupInterface):
ids = self._table.lookup(values) ids = self._table.lookup(values)
buckets = math_ops.add(buckets, self._table.size()) buckets = math_ops.add(buckets, self._table.size())
is_id_non_default = math_ops.not_equal(ids, self._table.default_value) is_id_non_default = math_ops.not_equal(ids, self._table.default_value)
ids = array_ops.where(is_id_non_default, ids, buckets) ids = array_ops.where_v2(is_id_non_default, ids, buckets)
else: else:
ids = buckets ids = buckets
if isinstance(keys, sparse_tensor.SparseTensor): if isinstance(keys, sparse_tensor.SparseTensor):
@ -1199,7 +1199,7 @@ class StaticVocabularyTable(LookupInterface):
ids = self._table.lookup(values) ids = self._table.lookup(values)
buckets = math_ops.add(buckets, self._table.size()) buckets = math_ops.add(buckets, self._table.size())
is_id_non_default = math_ops.not_equal(ids, self._table.default_value) is_id_non_default = math_ops.not_equal(ids, self._table.default_value)
ids = array_ops.where(is_id_non_default, ids, buckets) ids = array_ops.where_v2(is_id_non_default, ids, buckets)
else: else:
ids = buckets ids = buckets
if isinstance(keys, sparse_tensor.SparseTensor): if isinstance(keys, sparse_tensor.SparseTensor):