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
PiperOrigin-RevId: 324101079
Change-Id: I1a80a6a8986fefadf71fd870b84f04968a95ab1b
diff --git a/tensorflow/python/keras/engine/input_layer.py b/tensorflow/python/keras/engine/input_layer.py
index 75cf496..4818c5c 100644
--- a/tensorflow/python/keras/engine/input_layer.py
+++ b/tensorflow/python/keras/engine/input_layer.py
@@ -164,7 +164,8 @@
else:
raise_eager_tensor_error = False
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
else:
if not tf_utils.is_symbolic_tensor(input_tensor):