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

Work around the outside compilation / resource variable issue.

Browse Source 
PiperOrigin-RevId: 335441457
Change-Id: Id9055f26d786f8d9936edc0312f73e86a420e56a
This commit is contained in:
Mehmet Deveci 2020-10-05 10:08:48 -07:00 committed by TensorFlower Gardener
parent f9c1d327a3
commit ab1478d380
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/tensor_tracer.py
View File

@ -1494,7 +1494,7 @@ class TensorTracer(object):
flush_op = tpu.outside_compilation(
_flush_fun, cache_val, self._replica_id,
training_util.get_or_create_global_step())
array_ops.identity(training_util.get_or_create_global_step()))
else:
flush_op = _flush_fun(cache_val, self._replica_id,
training_util.get_or_create_global_step())