Skip None when inference shape when layer has None Input.

PiperOrigin-RevId: 820426089

Author: tensorflower-gardener

tf_keras/engine/base_layer_v1.py

@@ -2186,8 +2186,10 @@ def _maybe_build(self, inputs):
                 else:
                     self._set_dtype_policy(policy.Policy(dtype))
             input_shapes = None
-            if all(hasattr(x, "shape") for x in input_list):
-                input_shapes = tf.nest.map_structure(lambda x: x.shape, inputs)
+            if any(hasattr(x, "shape") for x in input_list):
+                input_shapes = tf.nest.map_structure(
+                    lambda x: x.shape if hasattr(x, 'shape') else None, inputs
+                )
             # Only call `build` if the user has manually overridden the build
             # method.
             if not hasattr(self.build, "_is_default"):