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

To limit behavior changes from the KerasTensors refactor, make sure the "EagerTensor cannot be passed to a Keras input" only triggers for actual eagertensors, not symbolic graph tensors

Browse Source 
PiperOrigin-RevId: 324101079
Change-Id: I1a80a6a8986fefadf71fd870b84f04968a95ab1b
This commit is contained in:
Tomer Kaftan 2020-07-30 15:56:30 -07:00 committed by TensorFlower Gardener
parent f80ef4b17e
commit b92c68721b
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/keras/engine/input_layer.py
View File

@ -164,7 +164,8 @@ class InputLayer(base_layer.Layer):
else: else:
raise_eager_tensor_error = False raise_eager_tensor_error = False
if keras_tensor.keras_tensors_enabled(): if keras_tensor.keras_tensors_enabled():
if not isinstance(input_tensor, keras_tensor.KerasTensor): if (not isinstance(input_tensor, keras_tensor.KerasTensor) and
not tf_utils.is_symbolic_tensor(input_tensor)):
raise_eager_tensor_error = True raise_eager_tensor_error = True
else: else:
if not tf_utils.is_symbolic_tensor(input_tensor): if not tf_utils.is_symbolic_tensor(input_tensor):