From 64f965e5aaf2d15d7011b99034dc68775d0186c9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ho=C3=ABl=20Bagard?= <34478245+hoel-bagard@users.noreply.github.com> Date: Mon, 19 Jan 2026 17:23:31 +0900 Subject: [PATCH] `tensorflow`: add `Activation`, `GlobalAveragePooling2D` and `MaxPool2D` layers (#15302) --- .../tensorflow/keras/layers/__init__.pyi | 44 +++++++++++++++++++ 1 file changed, 44 insertions(+) diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi index bb98d35d2..9764b04db 100644 --- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi +++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi @@ -443,4 +443,48 @@ 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: ... + +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: ... + +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