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

Replace set, dict with ObjectIdentityDict/Set to prepare for eq implementation

Browse Source 
PiperOrigin-RevId: 261786323
This commit is contained in:
Yanhua Sun 2019-08-05 15:45:17 -07:00 committed by TensorFlower Gardener
parent bc5a87b03e
commit 0094a421fc
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/eager/wrap_function.py
View File

@ -153,7 +153,7 @@ def _lift_unlifted_variables(graph, variable_holder):
ops.GraphKeys.LOCAL_VARIABLES)
existing_captures = object_identity.ObjectIdentitySet(
graph.internal_captures)
lifted_variables = {}
lifted_variables = object_identity.ObjectIdentityDictionary()
def _should_lift_variable(v):
return ((v._in_graph_mode # pylint: disable=protected-access