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

Only pre-read variables in tf.identity when executing eagerly

Browse Source 
Not necessary when graph building, and someone somewhere wrote a test which relies on it not happening...

PiperOrigin-RevId: 263207114
This commit is contained in:
Allen Lavoie 2019-08-13 13:49:58 -07:00 committed by TensorFlower Gardener
parent a2a84a8d59
commit bdfdec06d5
1 changed files with 4 additions and 2 deletions

6
tensorflow/python/ops/array_ops.py
View File

@ -196,8 +196,10 @@ def identity(input, name=None): # pylint: disable=redefined-builtin
Returns:
A `Tensor`. Has the same type as `input`.
"""
# Make sure we get an input with handle data attached from resource variables.
input = ops.convert_to_tensor(input)
if context.executing_eagerly() and not hasattr(input, "graph"):
# Make sure we get an input with handle data attached from resource
# variables. Variables have correct handle data when graph building.
input = ops.convert_to_tensor(input)
ret = gen_array_ops.identity(input, name=name)
# Propagate handle data for happier shape inference for resource variables.
if hasattr(input, "_handle_data"):