tf-keras-nightly 2.20.0.dev2025051109__py3-none-any.whl → 2.21.0.dev2025123010__py3-none-any.whl

tf_keras/src/saving/legacy/saved_model/save_impl.py

@@ -219,7 +219,11 @@ def wrap_layer_functions(layer, serialization_cache):
     with tracing_scope():
         call_collection.trace_with_input_signature()
         with base_layer_utils.call_context().enter(
-            layer, inputs=None, build_graph=True, training=None, saving=True
+            layer,
+            inputs=None,
+            build_graph=True,
+            call_context_args={},
+            saving=True,
         ):
             for fn in fns.values():
                 if fn is not None and not isinstance(fn, LayerCall):
@@ -515,19 +519,28 @@ class LayerCallCollection:
             else:
                 add_trace_to_queue(fn, args, kwargs)
 
-    def training_arg_was_passed(self, args, kwargs):
+    def arg_was_passed(self, arg_name, args, kwargs):
+        """Returns True if the argument was passed to the call function."""
         return self._call_spec.arg_was_passed(
-            "training", args, kwargs, inputs_in_args=True
+            arg_name, args, kwargs, inputs_in_args=True
         )
 
-    def get_training_arg_value(self, args, kwargs):
+    def training_arg_was_passed(self, args, kwargs):
+        """Returns True if the training arg was passed to the call function."""
+        return self.arg_was_passed("training", args, kwargs)
+
+    def get_arg_value(self, arg_name, args, kwargs):
+        """Returns the value of the given argument or None if not found."""
         try:
             return self._call_spec.get_arg_value(
-                "training", args, kwargs, inputs_in_args=True
+                arg_name, args, kwargs, inputs_in_args=True
             )
-        except KeyError:  # Training is not in args or kwargs.
+        except KeyError:  # Arg not found in args or kwargs.
             return None
 
+    def get_training_arg_value(self, args, kwargs):
+        return self.get_arg_value("training", args, kwargs)
+
     def get_input_arg_value(self, args, kwargs):
         return self._call_spec.get_arg_value(
             self._input_arg_name, args, kwargs, inputs_in_args=True
@@ -613,20 +626,23 @@ def layer_call_wrapper(call_collection, method, name):
     def wrapper(*args, **kwargs):
         """Calls method within call context."""
         layer = call_collection.layer
-        training = None
+        propagated = {"training": None}
         inputs = _filtered_inputs([args, kwargs])
 
-        if (args or kwargs) and call_collection.training_arg_was_passed(
-            args, kwargs
-        ):
-            training = call_collection.get_training_arg_value(args, kwargs)
+        for context_arg in layer._call_context_args:
+            if (args or kwargs) and call_collection.arg_was_passed(
+                context_arg, args, kwargs
+            ):
+                propagated[context_arg] = call_collection.get_arg_value(
+                    context_arg, args, kwargs
+                )
 
         original_losses = _reset_layer_losses(layer)
         with base_layer_utils.call_context().enter(
             layer,
             inputs=inputs,
             build_graph=False,
-            training=training,
+            call_context_args=propagated,
             saving=True,
         ):
             with autocast_variable.enable_auto_cast_variables(

tf_keras/src/saving/legacy/saving_utils.py

@@ -138,12 +138,24 @@ def trace_model_call(model, input_signature=None):
     @tf.function
     def _wrapped_model(*args, **kwargs):
         """A concrete tf.function that wraps the model's call function."""
+        call_context = base_layer_utils.call_context()
+
+        args, kwargs, propagated = model._get_propagated_call_context_arguments(
+            args, kwargs, call_context, model._call_context_args
+        )
+
         (args, kwargs,) = model._call_spec.set_arg_value(
             "training", False, args, kwargs, inputs_in_args=True
         )
 
-        with base_layer_utils.call_context().enter(
-            model, inputs=None, build_graph=False, training=False, saving=True
+        propagated["training"] = False
+
+        with call_context.enter(
+            model,
+            inputs=None,
+            build_graph=False,
+            call_context_args=propagated,
+            saving=True,
         ):
             outputs = model(*args, **kwargs)
 

tf_keras/src/saving/saving_lib.py

@@ -639,7 +639,7 @@ class NpzIOStore:
                 self.f = archive.open(root_path, mode="r")
             else:
                 self.f = open(root_path, mode="rb")
-            self.contents = np.load(self.f, allow_pickle=True)
+            self.contents = np.load(self.f, allow_pickle=False)
 
     def make(self, path):
         if not path:

tf_keras/src/utils/layer_utils.py

@@ -775,11 +775,13 @@ class CallFunctionSpec:
     """Caches the spec and provides utilities for handling call function
     args."""
 
-    def __init__(self, full_argspec):
+    def __init__(self, full_argspec, call_context_args=set()):
         """Initialies a `CallFunctionSpec`.
 
         Args:
           full_argspec: the FullArgSpec of a call function of a layer.
+          call_context_args: The set of call-context arguments registered
+            with to the current layer.
         """
         self._full_argspec = full_argspec
 
@@ -797,6 +799,18 @@ class CallFunctionSpec:
             "mask" in self._arg_names or call_accepts_kwargs
         )
 
+        # Track the set of call-context arguments that the current layer's
+        # `call` method accepts.
+        self._expected_context_args = set()
+        self._update_call_context_arguments(call_context_args)
+
+        self._context_arg_defaults = dict()
+        self._update_call_context_argument_defaults(call_context_args)
+
+    def _update_call_context_argument_defaults(self, context_args):
+        """Updates the set of call-context argument defaults for the current
+        layer's `call` method.
+        """
         call_fn_defaults = self._full_argspec.defaults or []
         defaults = dict()
         # The call arg defaults are an n-tuple of the last n elements of the
@@ -806,7 +820,21 @@ class CallFunctionSpec:
         # The default training arg will be any (non-None) default specified in
         # the method signature, or None if no value is specified.
         defaults.update(self._full_argspec.kwonlydefaults or {})
-        self._default_training_arg = defaults.get("training")
+
+        for arg in context_args:
+            self._context_arg_defaults[arg] = defaults.get(arg)
+
+    def _update_call_context_arguments(self, context_args):
+        """Updates the set of call-context arguments that the current layer's
+        `call` method accepts.
+        """
+        call_accepts_kwargs = self._full_argspec.varkw is not None
+        args_to_add = {
+            arg
+            for arg in context_args
+            if call_accepts_kwargs or arg in self._arg_names
+        }
+        self._expected_context_args.update(args_to_add)
 
     @property
     def full_argspec(self):
@@ -843,6 +871,16 @@ class CallFunctionSpec:
     def expects_training_arg(self, value):
         self._expects_training_arg = value
 
+    @property
+    def expected_context_args(self):
+        """The set of call-context arguments that the current layer's
+        `call` method accepts."""
+        return self._expected_context_args
+
+    @expected_context_args.setter
+    def expected_context_args(self, value):
+        self._expected_context_args = value
+
     @property
     def expects_mask_arg(self):
         """Whether the call function uses `mask` as a parameter."""
@@ -855,7 +893,11 @@ class CallFunctionSpec:
     @property
     def default_training_arg(self):
         """The default value given to the "training" argument."""
-        return self._default_training_arg
+        return self.get_context_arg_default("training")
+
+    def get_context_arg_default(self, arg_name):
+        """The default value given to the call context arguments."""
+        return self._context_arg_defaults.get(arg_name, None)
 
     def arg_was_passed(self, arg_name, args, kwargs, inputs_in_args=False):
         """Returns true if argument is present in `args` or `kwargs`.

tf_keras/src/utils/metrics_utils.py

@@ -237,6 +237,7 @@ class AUCCurve(Enum):
 
     ROC = "ROC"
     PR = "PR"
+    PR_GAIN = "PR_GAIN"
 
     @staticmethod
     def from_str(key):
@@ -244,10 +245,12 @@ class AUCCurve(Enum):
             return AUCCurve.PR
         elif key in ("roc", "ROC"):
             return AUCCurve.ROC
+        elif key in ("pr_gain", "prgain", "PR_GAIN", "PRGAIN"):
+            return AUCCurve.PR_GAIN
         else:
             raise ValueError(
                 f'Invalid AUC curve value: "{key}". '
-                'Expected values are ["PR", "ROC"]'
+                'Expected values are ["PR", "ROC", "PR_GAIN"]'
             )
 
 

{tf_keras_nightly-2.20.0.dev2025051109.dist-info → tf_keras_nightly-2.21.0.dev2025123010.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tf_keras-nightly
-Version: 2.20.0.dev2025051109
+Version: 2.21.0.dev2025123010
 Summary: Deep learning for humans.
 Home-page: https://keras.io/
 Download-URL: https://github.com/keras-team/tf-keras/tags
@@ -26,7 +26,7 @@ Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: tf-nightly~=2.20.0.dev
+Requires-Dist: tf-nightly~=2.21.0.dev
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{tf_keras_nightly-2.20.0.dev2025051109.dist-info → tf_keras_nightly-2.21.0.dev2025123010.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tf_keras/__init__.py,sha256=mXiuxjUXESvbNgowgb81sFtioOFLQZf7Z0pEE-kZYbE,911
+tf_keras/__init__.py,sha256=o1qbOe_ffzZ4LRxDpb9Ga8DP1A89Fhr9103F88EajA4,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
@@ -201,12 +201,12 @@ tf_keras/preprocessing/image/__init__.py,sha256=H6rbMLtlGIy_jBLCSDklVTMXUjEUe8KQ
 tf_keras/preprocessing/sequence/__init__.py,sha256=Zg9mw0TIRIc-BmVtdXvW3jdIQo05VHZX_xmqZDMuaik,285
 tf_keras/preprocessing/text/__init__.py,sha256=1yQd-VZD6SjnEpPyBFLucYMxu9A5DnAnIec2tba9zQk,329
 tf_keras/protobuf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tf_keras/protobuf/projector_config_pb2.py,sha256=GHQfZbNY6IgeVYvL1A9o5ET7EiG-jv--zhbAJg9Ez3k,1821
-tf_keras/protobuf/saved_metadata_pb2.py,sha256=K4ROX6DQeyFej5TBrUvfY7e_gzpQuCRuiuiVgk3ehhg,1585
-tf_keras/protobuf/versions_pb2.py,sha256=HP6fzinb4-KIEZaINXIAe-BpxQnGROxrxECgGcpcvFE,1119
+tf_keras/protobuf/projector_config_pb2.py,sha256=Qn_IXcjH7GYRpehPH4hznWxGDPn_No8kLTJHWkkCAu4,2208
+tf_keras/protobuf/saved_metadata_pb2.py,sha256=nI5xD26EgxUpHE2JyJsyDEHi6O8Tunue9j_JN9BUX08,1946
+tf_keras/protobuf/versions_pb2.py,sha256=0XsJrhgwioUlNc4hdeJ8tee3tWSTvZnLEwJd--wAVr4,1450
 tf_keras/regularizers/__init__.py,sha256=D6TnroEDjnyP79TY_624g2DToxVWuKzuaiBAn_gUQaY,634
 tf_keras/saving/__init__.py,sha256=Xo0imlDhiYV7Rowy8BjMwrFJuAB8h2DdIuVcxvaeEa0,681
-tf_keras/src/__init__.py,sha256=0_kJChFpSVveEmORgCyrhovmeKxMEzS78Hh1b4Egy18,1502
+tf_keras/src/__init__.py,sha256=p7hwxQsHJzS7Qq-OTj-gcJgsGn9bw5FN9IQ03yQO_Ls,1502
 tf_keras/src/activations.py,sha256=QNTCdIuNGww5BPwkkjkaNZf4j09m27Nqi-r4aTBOxnk,22630
 tf_keras/src/backend.py,sha256=-_L2aB0n9ZkFsRoivRX-BjGxBpJG27PunUm9DzN0fvU,248509
 tf_keras/src/backend_config.py,sha256=DaKkQg6jLmzR2GtgjNxwFoHuTXwVcAzx_Hx8XgAKPNs,4516
@@ -271,16 +271,16 @@ 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=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_layer.py,sha256=r_0g7rX2WA7-fThh3sjFqNZgSTEuR1CTsW4FrZMMI6s,162338
+tf_keras/src/engine/base_layer_utils.py,sha256=AFjqwXM-WShf0dfsyIotlXYIRJlqYyjQhAf50xZgyos,36166
+tf_keras/src/engine/base_layer_v1.py,sha256=MMfdUKB8tjbjjX9Pj5b6E5XgrM-BnVx0ilSndcR_3QA,102875
 tf_keras/src/engine/base_preprocessing_layer.py,sha256=xne5VVtj9_IE1_cjh-kaPk-utoMY7mYwTOcgybFfY34,12650
 tf_keras/src/engine/compile_utils.py,sha256=F6KxbaXnppns5XCOJl8wzsiQ1riEp43s0G0SWsWAUE0,31757
-tf_keras/src/engine/data_adapter.py,sha256=UqYJBUDiS-vyu7euVYxQrXw0U9-piO7SwTetkGBSMwg,71654
-tf_keras/src/engine/functional.py,sha256=igoSYLFSVYvZ8EUjwbowrDyYywI-V3h9ypeybIN_Jr4,70182
+tf_keras/src/engine/data_adapter.py,sha256=N5UV4KYF-F7YJdB5kHR8pACSrFlSqQte_DsmA8Ksa6Y,72257
+tf_keras/src/engine/functional.py,sha256=ojvj0DkGrnqd16dEIqj2AgFlmxB2s6e-3qjs78fN85E,71006
 tf_keras/src/engine/functional_utils.py,sha256=5creFfo9UoG5OLJgkcw9gsfT-qch-RamT5IsU8675rU,11048
-tf_keras/src/engine/input_layer.py,sha256=QVAA9ZrhfUlcx0Tj_UuNF3t1nxYrhyks6vDJJeb18W8,18258
-tf_keras/src/engine/input_spec.py,sha256=W3mojApaM_lN8Vr2MCvddE8RhHckxwObzpySPm16oEM,12076
+tf_keras/src/engine/input_layer.py,sha256=SRoRYG_PElMRMAypjeQRZ_0Ub4tu5jdRc6ASL59BAsY,18650
+tf_keras/src/engine/input_spec.py,sha256=H2U8yNz7eabUozm4QAcL3XcQoxj6iAwvK6ecU7w8O5g,12455
 tf_keras/src/engine/keras_tensor.py,sha256=rmIyf-sMKzGAMXzob0hCTZ3qA4JBYyIM85XUdmOPmqQ,28858
 tf_keras/src/engine/node.py,sha256=mevKNFEtzeVbwLRuwB7sMzQGKt6ppIxLmMcfQMzu8N8,14254
 tf_keras/src/engine/partial_batch_padding_handler.py,sha256=TNZvGXL-fvmZLLHIMPX_hy0w9LT8W52DHW7ZtnEvBvI,4325
@@ -314,13 +314,13 @@ tf_keras/src/layers/activation/elu.py,sha256=n-WAE6NjC9mbqcV7Kxgpt8tTbvwCQIGsoCV
 tf_keras/src/layers/activation/leaky_relu.py,sha256=cJmpwgg4KEu--iK9gFuJT7uEGpDArB8q-XNBmJfC7_U,2618
 tf_keras/src/layers/activation/prelu.py,sha256=D2yhneQrYQP6aHSK8nvnMKa1hIeuPZO_XCB2Cu9Cl4Y,4440
 tf_keras/src/layers/activation/relu.py,sha256=JklQuReRiR3huAGr3QRtuGL0URpdspDFzBNjZgv0HDw,4281
-tf_keras/src/layers/activation/softmax.py,sha256=G6MfTCogGTKwyP7b6ByxeIHFNQtUKgrZXB8MP9hNstQ,4105
+tf_keras/src/layers/activation/softmax.py,sha256=0g8uN5N8QDW8lj6nGabR-EBk58njbiNdhDzglv9rxXU,4861
 tf_keras/src/layers/activation/thresholded_relu.py,sha256=rQLn9cr-w6hVJET2mS7OIQ9diiUiqUrX4CysXKNYbmg,2503
 tf_keras/src/layers/attention/__init__.py,sha256=6HjPSyLhs_bf4erT65KyhSCHQF7WeWZe9YTH7iW6Nek,945
 tf_keras/src/layers/attention/additive_attention.py,sha256=jie0cAXJEjU4xXK_Ur1SrEL9RqDIIAPyaAkK8O71TEs,7485
 tf_keras/src/layers/attention/attention.py,sha256=TCnoOWAfh6i275TvudxyjosczBmL_zz9ByEUi-xXkAU,8682
 tf_keras/src/layers/attention/base_dense_attention.py,sha256=cEzBldjwQfuJfNZRimW5s-NqyENU2-lmqaNNxAGxhKw,10856
-tf_keras/src/layers/attention/multi_head_attention.py,sha256=05RC-2BSmCcBFtVY2loQPeiMYp8XArmbvovPl8kpiEA,30279
+tf_keras/src/layers/attention/multi_head_attention.py,sha256=FQX0YtXRy5kg8OlShA7cp2kfczzeWb9Oj3tbzkukLRw,30618
 tf_keras/src/layers/convolutional/__init__.py,sha256=U-4tja5JhSUva2G9uMmsZyZty2N2N9jT6EJRu5HAo-Y,3355
 tf_keras/src/layers/convolutional/base_conv.py,sha256=jvm4elEyIVSNfYZxh4inzQ1Q2CKS_f8VawvXMIJFSC4,17574
 tf_keras/src/layers/convolutional/base_depthwise_conv.py,sha256=SVgR2Y8dpeX4eDEF1e0UY0Mxh4A47eGHhJCQ1peGwNQ,9661
@@ -343,7 +343,7 @@ tf_keras/src/layers/core/embedding.py,sha256=iOdkBiP1IzwOVPjsKWA54NXrlk5KgJ0DfQ8
 tf_keras/src/layers/core/identity.py,sha256=yj5cWlUTlYq_J_ZQb1iLzM0bqaM4V6TXVwM4iuBFp9U,1301
 tf_keras/src/layers/core/lambda_layer.py,sha256=QzetX-lV9ybonQKg_6QzSm8w9Vkq8CPAM4BcAke7CZk,16481
 tf_keras/src/layers/core/masking.py,sha256=19p6HYGlKdUfQnelsAoee6wf87fWx67NSGinyjagNc4,3340
-tf_keras/src/layers/core/tf_op_layer.py,sha256=4WDRrT8dVwnD7avcWvMCk9mnGwfHcaN3Dmhf7CBeqzQ,21066
+tf_keras/src/layers/core/tf_op_layer.py,sha256=R6dFECVkPbmKi1nQVcxJy5lNxSVwiMlaWXB7j0PjI7Q,21320
 tf_keras/src/layers/experimental/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 tf_keras/src/layers/experimental/dynamic_embedding.py,sha256=KuVIawm3avPEa5c2IDOyBH14xiU5bYbPqcm_HugfWYA,10730
 tf_keras/src/layers/experimental/dynamic_lookup.py,sha256=CMNOaxAIkB1ChPcusuymhLAYTvobEbCBli6YkuWw8RE,13720
@@ -402,27 +402,7 @@ tf_keras/src/layers/preprocessing/preprocessing_utils.py,sha256=OR8NDGv8foDT2Ngv
 tf_keras/src/layers/preprocessing/string_lookup.py,sha256=2yqsgps42qMd6MB6vwBevionU7dh77OQdLburmn90b0,19179
 tf_keras/src/layers/preprocessing/text_vectorization.py,sha256=mL6sHm3TPXKg8q51vEWyo7LYKyiEoQFzm7GUkrSS-6E,30467
 tf_keras/src/layers/preprocessing/benchmarks/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-tf_keras/src/layers/preprocessing/benchmarks/bucketized_column_dense_benchmark.py,sha256=ZKFxRPRDx9VYUzu3k42DO2hrN9Ve9UNLPYEraN3BU94,2845
-tf_keras/src/layers/preprocessing/benchmarks/category_encoding_benchmark.py,sha256=IEdxK6eQa1YdxgmOQ13YBeJ94afFWfGazAO6NvfxJ5w,2949
-tf_keras/src/layers/preprocessing/benchmarks/category_hash_dense_benchmark.py,sha256=yoGE5ofB7fspimQ1ImShs5KguNGUQ_JpsGYPLZS1gpQ,2809
-tf_keras/src/layers/preprocessing/benchmarks/category_hash_varlen_benchmark.py,sha256=5b80c35WGpWEXgv2lutqVVuS52mWiD6Cyw1ZA6KkseU,2723
-tf_keras/src/layers/preprocessing/benchmarks/category_vocab_file_dense_benchmark.py,sha256=SPmA9yXH3dr6uHfs1IsAkrjNo02YgyfmWrt24pl6ROs,3588
-tf_keras/src/layers/preprocessing/benchmarks/category_vocab_file_varlen_benchmark.py,sha256=JAM0X1lBkZd7KYtBFaBP2HfxxB3Uj7Ik7WeFhajbwNo,3437
-tf_keras/src/layers/preprocessing/benchmarks/category_vocab_list_dense_benchmark.py,sha256=y0RR1TMq5PUv4Jlh7jMmQrJWsjDtgDivsqTeEMi6ovI,2863
-tf_keras/src/layers/preprocessing/benchmarks/category_vocab_list_indicator_dense_benchmark.py,sha256=lyfRE8NP3gLfTDnIzucPqjMiLAOCEUS-pSwa1f7EXLM,3169
-tf_keras/src/layers/preprocessing/benchmarks/category_vocab_list_indicator_varlen_benchmark.py,sha256=Ebx54Qo5ec-Sys5bPhp1KaVtmWsQxpNotzxxpxtfBPg,3101
-tf_keras/src/layers/preprocessing/benchmarks/category_vocab_list_varlen_benchmark.py,sha256=WkDOo5borQYk78xKbnsh7tcEZyjrDEGb7NnTkYzoM18,2795
-tf_keras/src/layers/preprocessing/benchmarks/discretization_adapt_benchmark.py,sha256=UD48alO_v-Vb8naluZtPozU7U4Oy-1WPSV1oqhzl-Yk,3776
-tf_keras/src/layers/preprocessing/benchmarks/embedding_dense_benchmark.py,sha256=PB7D3pFmVxlxZ4tKO7N-NB-wfJ0KY8B4RpAy_BZG01A,2836
-tf_keras/src/layers/preprocessing/benchmarks/embedding_varlen_benchmark.py,sha256=-fif0N3JPQT9fIwmpj-XE2eJif21sK2TsC6fri7ZuWI,2831
 tf_keras/src/layers/preprocessing/benchmarks/feature_column_benchmark.py,sha256=cSSHeEGH1dhxR3UJiCFZUgFeRZHd25eDUMRmKE140is,4814
-tf_keras/src/layers/preprocessing/benchmarks/hashed_crossing_benchmark.py,sha256=QV4n0f2j5b1Us-D2NHMA7WMRuUeMyiZpg-FAEopK0qs,2835
-tf_keras/src/layers/preprocessing/benchmarks/hashing_benchmark.py,sha256=wV16NUaNLfYZVwZCuMiX7JN9YDmbqyxaWHERo_uFJoE,3624
-tf_keras/src/layers/preprocessing/benchmarks/image_preproc_benchmark.py,sha256=x-XDwI75oIW3clnGOOmRG0Tb3hsQTx40bwxT7sj6CaE,5467
-tf_keras/src/layers/preprocessing/benchmarks/index_lookup_adapt_benchmark.py,sha256=LLv8vcdsphIBy5-owcABZdVSGSGMmQ7W-LmFTezO9Wc,4475
-tf_keras/src/layers/preprocessing/benchmarks/index_lookup_forward_benchmark.py,sha256=O4e0X-yLYWpfN2pX_WshN92ygw7XqlXZfgQjeO1WjuY,4941
-tf_keras/src/layers/preprocessing/benchmarks/normalization_adapt_benchmark.py,sha256=sB-Tcem8UdFGXnKx4HI4fLjTsIjaGJ2WAaphrxuItVc,4420
-tf_keras/src/layers/preprocessing/benchmarks/weighted_embedding_varlen_benchmark.py,sha256=Z5k0UaPM0-VfUw9tMv4_dEhsQNDODWlfNtsZ1RHFrFI,3324
 tf_keras/src/layers/regularization/__init__.py,sha256=9fIrtV8SwP1PG8BXfNrSP8rSyCdh4pPnV7hNvDbRysg,1369
 tf_keras/src/layers/regularization/activity_regularization.py,sha256=QxnBlnkHi2HZ2Pt-mX5WGiJWzljNQmh-X4La9f7XDGo,1942
 tf_keras/src/layers/regularization/alpha_dropout.py,sha256=JmMO6OHzpVtRS2Tl1fTslktQPM4MuN0ivNlCOUhH0VM,3800
@@ -454,7 +434,7 @@ tf_keras/src/layers/rnn/base_cudnn_rnn.py,sha256=cuPVg6r4L1pVWYTp3WFbJhikuIR2Vmg
 tf_keras/src/layers/rnn/base_rnn.py,sha256=I7mWl4KQC26gILDt9pZ9moZ81yM57lvci6hzJ9ROrxo,41968
 tf_keras/src/layers/rnn/base_wrapper.py,sha256=x4GANiXtmh9ztAFh7QtfbnQE76UVCGpaHp_XhrSs0Os,3159
 tf_keras/src/layers/rnn/bidirectional.py,sha256=JyZuBU0q2lt4augThwm8vyTvYwEJxyawsHmgNIul5vU,22670
-tf_keras/src/layers/rnn/cell_wrappers.py,sha256=T3FIiY9vIr0Or1N_SWNVnHR3LH6xnJ5DgNNYLk-sV6c,26874
+tf_keras/src/layers/rnn/cell_wrappers.py,sha256=fMGpdFFoRWRIuKz88NcnMvAtevv8OYHzxkF86Ltmwfk,27384
 tf_keras/src/layers/rnn/conv_lstm1d.py,sha256=suShze6ipNXabGlKJTxkOia17ZP4SeEei3Mi4F8lFOQ,8761
 tf_keras/src/layers/rnn/conv_lstm2d.py,sha256=myxOioB3yNn0L_-gMh0R41sb-MwTXO993lAT05_N0Zw,8874
 tf_keras/src/layers/rnn/conv_lstm3d.py,sha256=GT4OoPFtCr5xgaaqy3ezt5DyDu8Ut-wQEihCOHFk0D4,8969
@@ -483,7 +463,7 @@ tf_keras/src/legacy_tf_layers/variable_scope_shim.py,sha256=kGAFW03pVWSB1DhHvQ1W
 tf_keras/src/metrics/__init__.py,sha256=dM8S0ZhfiyPaXkdYuOSKvoytmYOkh8aYuJnpgUoT6vg,9699
 tf_keras/src/metrics/accuracy_metrics.py,sha256=RRQqyYZcVrEY2Pfc-OV6k3rYhv9ejSLJ9JbJzs_D5vk,17514
 tf_keras/src/metrics/base_metric.py,sha256=MCaI7Bx-kgs5udTRLvKMJ3SO90-GFs_9QMigrhkX9HQ,36498
-tf_keras/src/metrics/confusion_metrics.py,sha256=k548s2aYSeNfo3ROlK_Br6BDwKbE-Bqu4Hq0_HbLf7g,65973
+tf_keras/src/metrics/confusion_metrics.py,sha256=V1uNFUc1zyjxd-m-D83QhJ9bkbtPCfsXf3CROOPWmzs,68068
 tf_keras/src/metrics/f_score_metrics.py,sha256=3uxqH9NNqoKaGPz-R6eERA23bK1TabCXrsJUz2sbetU,12000
 tf_keras/src/metrics/hinge_metrics.py,sha256=QXtNdxE-IgZmdVQXIew_pN6X3aF9i7r7xirmb6oiOKA,4132
 tf_keras/src/metrics/iou_metrics.py,sha256=dUqZpOppIPj3aCtS25Hs6bvJoPHNnrtAChujoA-6bLQ,28530
@@ -498,7 +478,7 @@ tf_keras/src/mixed_precision/policy.py,sha256=1GWHp99dU0f6D0h_jIrSQkoLyIf0ClRJ0B
 tf_keras/src/mixed_precision/test_util.py,sha256=S4dDVLvFmv3OXvo-7kswO8MStwvTjP_caE3DrUhy9Po,8641
 tf_keras/src/models/__init__.py,sha256=VQ3cZve-CsmM_4CEi9q-V7m2qFO9HbdiO38mAR4dKdM,1823
 tf_keras/src/models/cloning.py,sha256=PHLTG0gSjvoKl8jxGaLCUq3ejK_o0PNA7gxSqxyoLBI,36839
-tf_keras/src/models/sharpness_aware_minimization.py,sha256=4nofg5_fbrRuGa5RAIQwJ-OL8eeiWg7jlNkMuJSCB_g,7301
+tf_keras/src/models/sharpness_aware_minimization.py,sha256=MArrweVZA85F1tPHZd06AVKpAdacaPplTz6eOS2XcRk,7795
 tf_keras/src/optimizers/__init__.py,sha256=lkPBfjJhWx_0nV8MrEmjWvJTGKutM1a9nIrB0ua0O-k,13044
 tf_keras/src/optimizers/adadelta.py,sha256=47HgdG0v-76B5htebkwl1OoryFPLO2kgk_CsYqgq7hU,6174
 tf_keras/src/optimizers/adafactor.py,sha256=_IYi6WMyXl4nPimr15nAPWvj6ZKcP7cESFsdpeabNKQ,8651
@@ -538,18 +518,17 @@ tf_keras/src/saving/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKV
 tf_keras/src/saving/object_registration.py,sha256=N8aV6eqREYjW2ueQpL3guYHyh5KXuun3DZAlmjfYrTA,7830
 tf_keras/src/saving/pickle_utils.py,sha256=5GtHzwNWVaYfZ-0zn69-zn2yv3R6JUwzHOOamnjP7r0,2605
 tf_keras/src/saving/saving_api.py,sha256=7uIWY4uwQMtETP5yRjZYPii-E-sWzSuK0ljVMfnZe_k,15119
-tf_keras/src/saving/saving_lib.py,sha256=Rk5rOvxEmCvwUlG3bS0QpOgtURayYmIAZbh2GeTuUOc,24272
+tf_keras/src/saving/saving_lib.py,sha256=Ik1m_D1G7_Z3KMipmy5Mi8jhyWtOnJ6QbFY-VV5A2R0,24273
 tf_keras/src/saving/serialization_lib.py,sha256=kX4qf_fRp4LySkH9FU37DMd0AXxiUrXKT-VLR3JPl7w,30152
 tf_keras/src/saving/legacy/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 tf_keras/src/saving/legacy/hdf5_format.py,sha256=IqFXHN96fuqKwu_akaqTyf9ISRPavP3Ahjydat948O4,42438
 tf_keras/src/saving/legacy/model_config.py,sha256=ZE6H_dKdmo2dlBWkr2nYO8SXcMEhshgza3sHPCpeu-k,4140
 tf_keras/src/saving/legacy/save.py,sha256=TdjiEamZ8MAsPAWsYMEtrdCRppbHcBIwJh9eVfdUS3k,23612
-tf_keras/src/saving/legacy/saving_utils.py,sha256=VTxnYFSWZ7m_40deANuWbtyhyXq0o0c5vLuBeCbgwi8,13745
+tf_keras/src/saving/legacy/saving_utils.py,sha256=0iXchqZQNw9s5kB9_7SIj2p3Qd_21jdfHJQ3b3YVWQs,14042
 tf_keras/src/saving/legacy/serialization.py,sha256=OrmHQPolQFsR-UCMxNTxkIFTKY4DKcAgMm1jdhF7TqU,22285
 tf_keras/src/saving/legacy/saved_model/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 tf_keras/src/saving/legacy/saved_model/base_serialization.py,sha256=dALR19_zt4c80zVw3yjCj9wfRoJufDjCrvkJyS82Dnk,5104
 tf_keras/src/saving/legacy/saved_model/constants.py,sha256=96ymvysCZ2Ru888YT_DEPMDgizHdDoBFGEOXsf-9AwE,1779
-tf_keras/src/saving/legacy/saved_model/create_test_saved_model.py,sha256=mS5jmCsDwUFUKr08G0tphPSA5ZAd7illNyj3QXKejOA,1040
 tf_keras/src/saving/legacy/saved_model/json_utils.py,sha256=WOyJaamx15lJ_V4XZSYM3RtuATa73uRNRM13o2s7yQ4,8071
 tf_keras/src/saving/legacy/saved_model/layer_serialization.py,sha256=FQwNk2XJq8dzgoSAWrmcabglZTA-6oDPtfeLiGrZO6A,8418
 tf_keras/src/saving/legacy/saved_model/load.py,sha256=wJUL0T4ZlUgxk3pZa_7E4NXEE3LXfjSKuGvTlrQchHc,57007
@@ -558,7 +537,7 @@ tf_keras/src/saving/legacy/saved_model/model_serialization.py,sha256=IxQ1TfBGagV
 tf_keras/src/saving/legacy/saved_model/network_serialization.py,sha256=ofbKN9V3syw0AQebgy2PlvaiAHi3SnBFTg-PUgclTng,1180
 tf_keras/src/saving/legacy/saved_model/order_preserving_set.py,sha256=zvNFzss8wSc0vngv74dNnQO_hxpxmEWWBBv1TTLsbPY,3250
 tf_keras/src/saving/legacy/saved_model/save.py,sha256=2-AaGFhFxzfZLkIW1qx9-rTcaZvYMFkQYP7ijfwA-ZI,6395
-tf_keras/src/saving/legacy/saved_model/save_impl.py,sha256=cWDBJ0uYfV1dNB5pdKFqDwYLQaaxWNeG8GSLiABAmec,29751
+tf_keras/src/saving/legacy/saved_model/save_impl.py,sha256=mcdNPwJYwzOsdSisgwkBEbnoSABEZdshN7BRmTttK2c,30420
 tf_keras/src/saving/legacy/saved_model/serialized_attributes.py,sha256=nlmtIzLUBGSQU6gDKcg4-ypSRX3RbS4vPmLIhG3HSbk,15009
 tf_keras/src/saving/legacy/saved_model/utils.py,sha256=2OCwun0U8nsZvxUbv7Toq2EeC1HU32LxnLDan8cw4Dc,9953
 tf_keras/src/testing_infra/__init__.py,sha256=yrmnTOUMQ09fOgD3PD4NjpaeKz2OXCUmmoExRWhg9AY,690
@@ -567,7 +546,6 @@ tf_keras/src/testing_infra/test_combinations.py,sha256=ETwFTN8eBAusQpqU7dg_Qckb1
 tf_keras/src/testing_infra/test_utils.py,sha256=SMEYejGPfYnZT2tVgzHL3gBHNGk6qcTu1qcZetHv870,40307
 tf_keras/src/tests/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 tf_keras/src/tests/get_config_samples.py,sha256=qz2SZb_JIW2NoTak9NphLJkDTgYmlQ5RNm64T9wQ6L8,15307
-tf_keras/src/tests/keras_doctest.py,sha256=qFPhxdstCjwGZw0JIKPMZ_PF-oBzEgP6EqZ9n_0mtio,4638
 tf_keras/src/tests/model_architectures.py,sha256=83-y4n0LtvpcpXPgawvPGIcvaqaPZ_XgOVEDRgycLmw,10830
 tf_keras/src/tests/model_subclassing_test_util.py,sha256=tMRAx38exGbDKEUd5kNDRn7Q-epiMPCAxdbAEGSCP6Y,5515
 tf_keras/src/utils/__init__.py,sha256=HDp6YtwWY9al-pSjokrgj_IzsFi36TWQVGJp3ibTlws,3129
@@ -585,9 +563,9 @@ tf_keras/src/utils/io_utils.py,sha256=XhCTkjwtfBc2hWSenjVdt0-2PsIc2bjJVWEP1880NU
 tf_keras/src/utils/keras_logging.py,sha256=Fv4eOMemx3Jg1hEdHIxx9GblG5YTnW1q1D1zLF3JxUE,882
 tf_keras/src/utils/kernelized_utils.py,sha256=s475SAos2zHQ1NT9AHZmbWUSahHKOhdctP6uIou0nRo,4517
 tf_keras/src/utils/kpl_test_utils.py,sha256=vnaJkySSTVhXsFEdDxNJArwXaah0yPNTK8o_3rYZvOE,7365
-tf_keras/src/utils/layer_utils.py,sha256=5SGCXE5Tc8QmCt7JHTbpVjYBkEMtdI7TQ5jC-9VLnCY,41764
+tf_keras/src/utils/layer_utils.py,sha256=cLKqiqJ2em16zZyXaXFErsL6yja28qE6kgPs2TTcdcY,43427
 tf_keras/src/utils/losses_utils.py,sha256=oPHJSNLY8U57ieQD59vnGHNavZpMpeTZtL7VIlDwwfM,16919
-tf_keras/src/utils/metrics_utils.py,sha256=feW5GoiznbQKkxmE3Url2nlXfWgvHMpJPXdGKCdiV_U,39803
+tf_keras/src/utils/metrics_utils.py,sha256=h4F4MGcHrpjthypj-nZ1n2szBrBZj4X0R9cEzMcx75w,39938
 tf_keras/src/utils/mode_keys.py,sha256=_QYq58qr_b-RhvMYBYnL47NkC0G1ng8NYcVnS_IYi-A,856
 tf_keras/src/utils/np_utils.py,sha256=4EZ58G1zThQfQEmMNBPnUYRszXRJoY4foxYhOGfS89s,4805
 tf_keras/src/utils/object_identity.py,sha256=HZEETVcCoBrnIFjnxmBhZaCKP9xQMv9rMr_ihlMveVs,6879
@@ -606,7 +584,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.20.0.dev2025051109.dist-info/METADATA,sha256=7pum9VVRblgMxzdMX-AnMCn92NgwN9jZ-9YV5vER-Dw,1857
-tf_keras_nightly-2.20.0.dev2025051109.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-tf_keras_nightly-2.20.0.dev2025051109.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
-tf_keras_nightly-2.20.0.dev2025051109.dist-info/RECORD,,
+tf_keras_nightly-2.21.0.dev2025123010.dist-info/METADATA,sha256=mMuSxuENfVse58L4xsGvhuHeJ_PYmN4HQTSCzIDhrQE,1857
+tf_keras_nightly-2.21.0.dev2025123010.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tf_keras_nightly-2.21.0.dev2025123010.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
+tf_keras_nightly-2.21.0.dev2025123010.dist-info/RECORD,,

{tf_keras_nightly-2.20.0.dev2025051109.dist-info → tf_keras_nightly-2.21.0.dev2025123010.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.4.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

tf_keras/src/layers/preprocessing/benchmarks/bucketized_column_dense_benchmark.py

@@ -1,85 +0,0 @@
-# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-"""Benchmark for KPL implementation of bucketized columns with dense inputs."""
-
-import numpy as np
-import tensorflow.compat.v2 as tf
-
-import tf_keras.src as keras
-from tf_keras.src.layers.preprocessing import discretization
-from tf_keras.src.layers.preprocessing.benchmarks import (
-    feature_column_benchmark as fc_bm,
-)
-
-# isort: off
-from tensorflow.python.eager.def_function import (
-    function as tf_function,
-)
-
-NUM_REPEATS = 10  # The number of times to run each benchmark.
-BATCH_SIZES = [32, 256]
-
-
-### KPL AND FC IMPLEMENTATION BENCHMARKS ###
-def embedding_varlen(batch_size, max_length):
-    """Benchmark a variable-length embedding."""
-    # Data and constants.
-    max_value = 25.0
-    bins = np.arange(1.0, max_value)
-    data = fc_bm.create_data(
-        max_length, batch_size * NUM_REPEATS, 100000, dtype=float
-    )
-
-    # TF-Keras implementation
-    model = keras.Sequential()
-    model.add(keras.Input(shape=(max_length,), name="data", dtype=tf.float32))
-    model.add(discretization.Discretization(bins))
-
-    # FC implementation
-    fc = tf.feature_column.bucketized_column(
-        tf.feature_column.numeric_column("data"), boundaries=list(bins)
-    )
-
-    # Wrap the FC implementation in a tf.function for a fair comparison
-    @tf_function()
-    def fc_fn(tensors):
-        fc.transform_feature(
-            tf.__internal__.feature_column.FeatureTransformationCache(tensors),
-            None,
-        )
-
-    # Benchmark runs
-    keras_data = {"data": data.to_tensor(default_value=0.0)}
-    k_avg_time = fc_bm.run_keras(keras_data, model, batch_size, NUM_REPEATS)
-
-    fc_data = {"data": data.to_tensor(default_value=0.0)}
-    fc_avg_time = fc_bm.run_fc(fc_data, fc_fn, batch_size, NUM_REPEATS)
-
-    return k_avg_time, fc_avg_time
-
-
-class BenchmarkLayer(fc_bm.LayerBenchmark):
-    """Benchmark the layer forward pass."""
-
-    def benchmark_layer(self):
-        for batch in BATCH_SIZES:
-            name = f"bucketized|dense|batch_{batch}"
-            k_time, f_time = embedding_varlen(batch_size=batch, max_length=256)
-            self.report(name, k_time, f_time, NUM_REPEATS)
-
-
-if __name__ == "__main__":
-    tf.test.main()
-

tf_keras/src/layers/preprocessing/benchmarks/category_encoding_benchmark.py

@@ -1,84 +0,0 @@
-# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-"""Benchmark for TF-Keras category_encoding preprocessing layer."""
-
-import time
-
-import numpy as np
-import tensorflow.compat.v2 as tf
-
-import tf_keras.src as keras
-from tf_keras.src.layers.preprocessing import category_encoding
-
-
-class BenchmarkLayer(tf.test.Benchmark):
-    """Benchmark the layer forward pass."""
-
-    def run_dataset_implementation(
-        self, output_mode, batch_size, sequence_length, max_tokens
-    ):
-        input_t = keras.Input(shape=(sequence_length,), dtype=tf.int32)
-        layer = category_encoding.CategoryEncoding(
-            max_tokens=max_tokens, output_mode=output_mode
-        )
-        _ = layer(input_t)
-
-        num_repeats = 5
-        starts = []
-        ends = []
-        for _ in range(num_repeats):
-            ds = tf.data.Dataset.from_tensor_slices(
-                tf.random.uniform(
-                    [batch_size * 10, sequence_length],
-                    minval=0,
-                    maxval=max_tokens - 1,
-                    dtype=tf.int32,
-                )
-            )
-            ds = ds.shuffle(batch_size * 100)
-            ds = ds.batch(batch_size)
-            num_batches = 5
-            ds = ds.take(num_batches)
-            ds = ds.prefetch(num_batches)
-            starts.append(time.time())
-            # Benchmarked code begins here.
-            for i in ds:
-                _ = layer(i)
-            # Benchmarked code ends here.
-            ends.append(time.time())
-
-        avg_time = np.mean(np.array(ends) - np.array(starts)) / num_batches
-        name = "category_encoding|batch_%s|seq_length_%s|%s_max_tokens" % (
-            batch_size,
-            sequence_length,
-            max_tokens,
-        )
-        self.report_benchmark(iters=num_repeats, wall_time=avg_time, name=name)
-
-    def benchmark_vocab_size_by_batch(self):
-        for batch in [32, 256, 2048]:
-            for sequence_length in [10, 1000]:
-                for num_tokens in [100, 1000, 20000]:
-                    self.run_dataset_implementation(
-                        output_mode="count",
-                        batch_size=batch,
-                        sequence_length=sequence_length,
-                        max_tokens=num_tokens,
-                    )
-
-
-if __name__ == "__main__":
-    tf.test.main()
-