Skip None when inference shape when layer has None Input.

PiperOrigin-RevId: 821862072

Author: tensorflower-gardener

tf_keras/engine/base_layer_v1.py

@@ -2186,8 +2186,8 @@ 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_utils.get_shapes(inputs)
             # Only call `build` if the user has manually overridden the build
             # method.
             if not hasattr(self.build, "_is_default"):