diff --git a/tensorflow/python/keras/engine/input_layer.py b/tensorflow/python/keras/engine/input_layer.py index 75cf4960d27..4818c5c59a7 100644 --- a/tensorflow/python/keras/engine/input_layer.py +++ b/tensorflow/python/keras/engine/input_layer.py @@ -164,7 +164,8 @@ class InputLayer(base_layer.Layer): 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):