commit 0d65cfaab050295c311d9f2fb28388435359db27
author Allen Wang <allencwang@google.com> Mon Oct 12 10:39:23 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Oct 12 11:00:39 2020 -0700
tree 123e858dfed177d7f4a59e4772d9851df1b80dfb
parent d855fc792a6718a2df89027011bd5cf411eccc89

Add an additional `NoneType` check when converting a traced tensor to a `KerasTensor`.

PiperOrigin-RevId: 336694750
Change-Id: Ic79bc3b46b81d4816a7108ed9b1aa426e7f4d3d5

tensorflow/python/keras/engine/keras_tensor.py

diff --git a/tensorflow/python/keras/engine/keras_tensor.py b/tensorflow/python/keras/engine/keras_tensor.py
index 0c2c8bf..0557520 100644
--- a/tensorflow/python/keras/engine/keras_tensor.py
+++ b/tensorflow/python/keras/engine/keras_tensor.py
@@ -170,7 +170,8 @@
       name = getattr(tensor, 'name', None)
       type_spec = type_spec_module.type_spec_from_value(tensor)
       inferred_value = None
-      if (type_spec.dtype == dtypes.int32 and type_spec.shape.rank < 2):
+      if (type_spec.dtype == dtypes.int32 and type_spec.shape.rank is not None
+          and type_spec.shape.rank < 2):
         # If this tensor might be representing shape information,
         # (dtype=int32, rank of 0 or 1, not too large to represent a shape)
         # we attempt to capture any value information tensorflow's