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

Avoid infinite loop in uses_keras_history

Browse Source 
PiperOrigin-RevId: 260934855
This commit is contained in:
Gaurav Jain 2019-07-31 08:52:02 -07:00 committed by TensorFlower Gardener
parent 170a95de67
commit 5820a245fa
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/keras/engine/base_layer_utils.py
View File

@ -310,6 +310,9 @@ def uses_keras_history(tensors):
while tensors_to_check: while tensors_to_check:
new_tensors_to_check = [] new_tensors_to_check = []
for tensor in tensors_to_check: for tensor in tensors_to_check:
if id(tensor) in checked_tensors:
continue
checked_tensors.add(id(tensor)) checked_tensors.add(id(tensor))
if getattr(tensor, '_keras_history_checked', None) is not None: if getattr(tensor, '_keras_history_checked', None) is not None:
@ -318,9 +321,7 @@ def uses_keras_history(tensors):
return True return True
try: try:
for t in tensor.op.inputs: new_tensors_to_check.extend(tensor.op.inputs)
if id(t) not in checked_tensors:
new_tensors_to_check.append(t)
except AttributeError: except AttributeError:
# In case `tensor` is a Variable created in an Eager context. # In case `tensor` is a Variable created in an Eager context.
pass pass