From 494a4b3a7571f5bf7e5797f4e55d55c082708295 Mon Sep 17 00:00:00 2001 From: Hoel Bagard Date: Mon, 19 Jan 2026 16:12:20 +0900 Subject: [PATCH 1/3] add tf.keras.layers.Activation --- .../tensorflow/keras/layers/__init__.pyi | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi index bb98d35d29de..6271b9c415d4 100644 --- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi +++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi @@ -443,4 +443,18 @@ class GaussianDropout(Layer[tf.Tensor, tf.Tensor]): name: str | None = None, ) -> None: ... +class Activation(Layer[tf.Tensor, tf.Tensor]): + def __init__( + self, + activation: _Activation = None, + *, + # **kwargs passed to Layer + # **kwargs passed to Layer + activity_regularizer: _Regularizer = None, + trainable: bool = True, + dtype: _LayerDtype | None = None, + autocast: bool = True, + name: str | None = None, + ) -> None: ... + def __getattr__(name: str): ... # incomplete module From 368b8d338063fe8a95cba25f6e6d88a5749dd72e Mon Sep 17 00:00:00 2001 From: Hoel Bagard Date: Mon, 19 Jan 2026 16:17:32 +0900 Subject: [PATCH 2/3] add tf.keras.layers.GlobalAveragePooling2D --- .../tensorflow/keras/layers/__init__.pyi | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi index 6271b9c415d4..3d501a517d64 100644 --- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi +++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi @@ -457,4 +457,18 @@ class Activation(Layer[tf.Tensor, tf.Tensor]): name: str | None = None, ) -> None: ... +class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]): + def __init__( + self, + data_format: Literal["channels_last", "channels_first"] | None = None, + keepdims: bool = False, + *, + # **kwargs passed to Layer + activity_regularizer: _Regularizer = None, + trainable: bool = True, + dtype: _LayerDtype | None = None, + autocast: bool = True, + name: str | None = None, + ) -> None: ... + def __getattr__(name: str): ... # incomplete module From fe09a9dcfe826ad3faa3130a96073734c7e2b332 Mon Sep 17 00:00:00 2001 From: Hoel Bagard Date: Mon, 19 Jan 2026 16:19:45 +0900 Subject: [PATCH 3/3] add tf.keras.layers.MaxPool2D --- .../tensorflow/keras/layers/__init__.pyi | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi index 3d501a517d64..9764b04dbe0e 100644 --- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi +++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi @@ -471,4 +471,20 @@ class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]): name: str | None = None, ) -> None: ... +class MaxPool2D(Layer[tf.Tensor, tf.Tensor]): + def __init__( + self, + pool_size: int | tuple[int, int] = (2, 2), + strides: int | tuple[int, int] | None = None, + padding: Literal["valid", "same"] = "valid", + data_format: Literal["channels_last", "channels_first"] | None = None, + *, + # **kwargs passed to Layer + activity_regularizer: _Regularizer = None, + trainable: bool = True, + dtype: _LayerDtype | None = None, + autocast: bool = True, + name: str | None = None, + ) -> None: ... + def __getattr__(name: str): ... # incomplete module