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

Use ObjectIdentitySet instead of set()

Browse Source 
PiperOrigin-RevId: 259070523
This commit is contained in:
Gaurav Jain 2019-07-19 17:54:01 -07:00 committed by TensorFlower Gardener
parent 3b2bf7e948
commit 6dcc61a0ae
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/framework/auto_control_deps.py
View File

@ -27,6 +27,7 @@ from tensorflow.python.ops import control_flow_ops
from tensorflow.python.ops import control_flow_util from tensorflow.python.ops import control_flow_util
from tensorflow.python.ops import tensor_array_ops from tensorflow.python.ops import tensor_array_ops
from tensorflow.python.util import nest from tensorflow.python.util import nest
from tensorflow.python.util import object_identity
from tensorflow.python.util import tf_decorator from tensorflow.python.util import tf_decorator
# Op types that should not run in program order, e.g. because they need to run # Op types that should not run in program order, e.g. because they need to run
@ -110,7 +111,7 @@ class AutomaticControlDependencies(object):
""" """
def __init__(self): def __init__(self):
self._returned_tensors = set() self._returned_tensors = object_identity.ObjectIdentitySet()
self.ops_which_must_run = set() self.ops_which_must_run = set()
def mark_as_return(self, tensor): def mark_as_return(self, tensor):