tf-keras-nightly 2.19.0.dev2025012310__py3-none-any.whl → 2.19.0.dev2025012510__py3-none-any.whl

tf_keras/__init__.py

@@ -27,4 +27,4 @@ from tf_keras.src.engine.sequential import Sequential
 from tf_keras.src.engine.training import Model
 
 
-__version__ = "2.19.0.dev2025012310"
+__version__ = "2.19.0.dev2025012510"

tf_keras/src/engine/base_layer.py

@@ -2321,7 +2321,7 @@ class Layer(tf.Module, version_utils.LayerVersionSelector):
         """
         input_shape = config["input_shape"]
         if input_shape is not None:
-            self.build(input_shape)
+            self.build(tf_utils.convert_shapes(input_shape, to_tuples=False))
 
     ############################################################################
     # Methods & attributes below are all private and only used by the framework.

tf_keras/src/engine/sequential.py

@@ -285,12 +285,16 @@ class Sequential(functional.Functional):
         ):
             # Determine whether the input shape is novel, i.e. whether the model
             # should be rebuilt.
-            input_shape = tuple(input_shape)
+            input_shape = tf_utils.convert_shapes(input_shape)
             if self._inferred_input_shape is None:
                 new_shape = input_shape
             else:
-                new_shape = relax_input_shape(
-                    self._inferred_input_shape, input_shape
+                new_shape = tf.nest.map_structure(
+                    _relax_input_shape,
+                    tf_utils.convert_shapes(
+                        self._inferred_input_shape, to_tuples=False
+                    ),
+                    tf_utils.convert_shapes(input_shape, to_tuples=False),
                 )
             if (
                 new_shape is not None
@@ -299,10 +303,13 @@ class Sequential(functional.Functional):
                 # A novel shape has been received: we need to rebuild the model.
                 # In case we are inside a graph function, we step out of it.
                 with tf.init_scope():
-                    inputs = input_layer.Input(
-                        batch_shape=new_shape,
-                        dtype=input_dtype,
-                        name=self.layers[0].name + "_input",
+                    inputs = tf.nest.map_structure(
+                        lambda s: input_layer.Input(
+                            batch_shape=tf_utils.convert_shapes(s),
+                            dtype=input_dtype,
+                            name=self.layers[0].name + "_input",
+                        ),
+                        tf_utils.convert_shapes(new_shape, to_tuples=False),
                     )
                     layer_input = inputs
                     created_nodes = set()
@@ -370,7 +377,7 @@ class Sequential(functional.Functional):
                 raise ValueError("You must provide an `input_shape` argument.")
             self._build_graph_network_for_inferred_shape(input_shape)
             if not self.built:
-                input_shape = tuple(input_shape)
+                input_shape = tf_utils.convert_shapes(input_shape)
                 self._build_input_shape = input_shape
                 super().build(input_shape)
         self.built = True
@@ -435,7 +442,8 @@ class Sequential(functional.Functional):
     def get_config(self):
         layer_configs = []
         serialize_obj_fn = serialization_lib.serialize_keras_object
-        if getattr(self, "use_legacy_config", None):
+        use_legacy_config = getattr(self, "use_legacy_config", False)
+        if use_legacy_config:
             serialize_obj_fn = legacy_serialization.serialize_keras_object
         for layer in super().layers:
             # `super().layers` include the InputLayer if available (it is
@@ -446,7 +454,11 @@ class Sequential(functional.Functional):
         config = training.Model.get_config(self)
         config["name"] = self.name
         config["layers"] = copy.deepcopy(layer_configs)
-        if not self._is_graph_network and self._build_input_shape is not None:
+        if (
+            use_legacy_config
+            and not self._is_graph_network
+            and self._build_input_shape
+        ):
             config["build_input_shape"] = self._build_input_shape
         return config
 
@@ -458,6 +470,7 @@ class Sequential(functional.Functional):
             layer_configs = config["layers"]
         else:
             name = None
+            build_input_shape = None
             layer_configs = config
         model = cls(name=name)
         for layer_config in layer_configs:
@@ -519,11 +532,15 @@ def _get_shape_tuple(t):
     return None
 
 
-def relax_input_shape(shape_1, shape_2):
+def _relax_input_shape(shape_1, shape_2):
     if shape_1 is None or shape_2 is None:
         return None
-    if len(shape_1) != len(shape_2):
+    if shape_1.rank is None or shape_2.rank is None:
+        return None
+    if shape_1.rank != shape_2.rank:
         return None
+    shape_1 = shape_1.as_list()
+    shape_2 = shape_2.as_list()
     return tuple(None if d1 != d2 else d1 for d1, d2 in zip(shape_1, shape_2))
 
 

{tf_keras_nightly-2.19.0.dev2025012310.dist-info → tf_keras_nightly-2.19.0.dev2025012510.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tf_keras-nightly
-Version: 2.19.0.dev2025012310
+Version: 2.19.0.dev2025012510
 Summary: Deep learning for humans.
 Home-page: https://keras.io/
 Download-URL: https://github.com/keras-team/tf-keras/tags

{tf_keras_nightly-2.19.0.dev2025012310.dist-info → tf_keras_nightly-2.19.0.dev2025012510.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tf_keras/__init__.py,sha256=mru8qED6C6luZYhhlk1TwIr0YB3w-Ck3RyUPcaFdNpg,911
+tf_keras/__init__.py,sha256=1tVgfYq8stDM5rLt1q0zp1G21GtnmKLu5jFtMBKREGw,911
 tf_keras/__internal__/__init__.py,sha256=OHQbeIC0QtRBI7dgXaJaVbH8F00x8dCI-DvEcIfyMsE,671
 tf_keras/__internal__/backend/__init__.py,sha256=LnMs2A6685gDG79fxqmdulIYlVE_3WmXlBTBo9ZWYcw,162
 tf_keras/__internal__/layers/__init__.py,sha256=F5SGMhOTPzm-PR44VrfinURHcVeQPIEdwnZlAkSTB3A,176
@@ -271,7 +271,7 @@ tf_keras/src/dtensor/lazy_variable.py,sha256=c3yylbga0se3Geflutss3fz5RzBYuY2vkU3
 tf_keras/src/dtensor/test_util.py,sha256=9QAbt44mlirdqwG2ertTsoXNKG2V4Z0bqJFxGdxy5BY,4572
 tf_keras/src/dtensor/utils.py,sha256=2TTSCEOA61Ia1FAPfQWJ2CRfiocBGUZreXH9UBFzFbk,6441
 tf_keras/src/engine/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-tf_keras/src/engine/base_layer.py,sha256=PaypExNUkGaWuIn9tiLN41AxSIhEXAals7b99Md0Yx8,156511
+tf_keras/src/engine/base_layer.py,sha256=H7TK3ezXORn1B79TZO3klFGE_hmXHAauS3h5k_8xJvA,156553
 tf_keras/src/engine/base_layer_utils.py,sha256=YMJF5sZJhFF_yzfqOtqi4YTsyUE2ZQ_cJJOIdXnuS2w,35795
 tf_keras/src/engine/base_layer_v1.py,sha256=cX-OCSNio3Tr2M6twr_PUgKulePLZDNW_4xLXjgYbN4,102700
 tf_keras/src/engine/base_preprocessing_layer.py,sha256=xne5VVtj9_IE1_cjh-kaPk-utoMY7mYwTOcgybFfY34,12650
@@ -285,7 +285,7 @@ tf_keras/src/engine/keras_tensor.py,sha256=rmIyf-sMKzGAMXzob0hCTZ3qA4JBYyIM85XUd
 tf_keras/src/engine/node.py,sha256=mevKNFEtzeVbwLRuwB7sMzQGKt6ppIxLmMcfQMzu8N8,14254
 tf_keras/src/engine/partial_batch_padding_handler.py,sha256=TNZvGXL-fvmZLLHIMPX_hy0w9LT8W52DHW7ZtnEvBvI,4325
 tf_keras/src/engine/saving.py,sha256=So_T5PRjCOLzpOPGHNBiTCOrNvqvvGNK8AwZBwFQCbs,853
-tf_keras/src/engine/sequential.py,sha256=UeW__ZHBL_lBgQfdyZZAOGW8KTjXy2dKsCu0IS3dR4Y,22974
+tf_keras/src/engine/sequential.py,sha256=YZaO53uU2XWmAa4wMjZ6XLiO2O5lPnYZ4IEeDCQAxqA,23675
 tf_keras/src/engine/training.py,sha256=eaEAV4OaV_0NmwIi_Upm-bZQKBzyCl_ilSFZ3Cinvqg,193232
 tf_keras/src/engine/training_arrays_v1.py,sha256=Fn_PY4_7miHhmkhXoNJS42LEpQDZ0VWOyNGv5m2DKNE,30903
 tf_keras/src/engine/training_distributed_v1.py,sha256=niN6TZ1DpXkWGUA7CWH6--VxFRLlX48RnqkGpHym4fE,32084
@@ -606,7 +606,7 @@ tf_keras/src/utils/legacy/__init__.py,sha256=EfMmeHYDzwvxNaktPhQbkTdcPSIGCqMhBND
 tf_keras/utils/__init__.py,sha256=b7_d-USe_EmLo02_P99Q1rUCzKBYayPCfiYFStP-0nw,2735
 tf_keras/utils/experimental/__init__.py,sha256=DzGogE2AosjxOVILQBT8PDDcqbWTc0wWnZRobCdpcec,97
 tf_keras/utils/legacy/__init__.py,sha256=7ujlDa5HeSRcth2NdqA0S1P2-VZF1kB3n68jye6Dj-8,189
-tf_keras_nightly-2.19.0.dev2025012310.dist-info/METADATA,sha256=1LYwnjGOiOrFwSf4rwSIie4xD_VEgxGbKFz5-eScQYg,1857
-tf_keras_nightly-2.19.0.dev2025012310.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tf_keras_nightly-2.19.0.dev2025012310.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
-tf_keras_nightly-2.19.0.dev2025012310.dist-info/RECORD,,
+tf_keras_nightly-2.19.0.dev2025012510.dist-info/METADATA,sha256=AhRzoJVwgSZYNQaok9WLXWL7_nTU-0qK--FmCGAITAk,1857
+tf_keras_nightly-2.19.0.dev2025012510.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tf_keras_nightly-2.19.0.dev2025012510.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
+tf_keras_nightly-2.19.0.dev2025012510.dist-info/RECORD,,