From 2283945db2ba186cc1e8a3550788e67c2191b7ae Mon Sep 17 00:00:00 2001 From: Daan Smienk Date: Fri, 24 Jan 2025 17:26:51 +0000 Subject: [PATCH 1/2] Add MaxPooling2D and GlobalAveragePooling2D --- .../tensorflow/keras/layers/__init__.pyi | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi index a2a2211b56cf..c4ae2c3f16f6 100644 --- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi +++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi @@ -442,4 +442,25 @@ class GaussianDropout(Layer[tf.Tensor, tf.Tensor]): name: str | None = None, ) -> None: ... +class MaxPooling2D(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: None | Literal["channels_last", "channels_first"] = None, + name: str | None = None, + ) -> None: ... + +MaxPool2D = MaxPooling2D + +class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]): + def __init__( + self, + data_format: None | Literal["channels_last", "channels_first"] = None, + keepdims: bool = False, + ) -> None: ... + +GlobalAvgPool2D = GlobalAveragePooling2D + def __getattr__(name: str) -> Incomplete: ... From 8ae91e8b4f2d141b8a568ba40b33e837b8a5a769 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 24 Jan 2025 17:30:40 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks --- stubs/tensorflow/tensorflow/keras/layers/__init__.pyi | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi index c4ae2c3f16f6..e2188abbd94b 100644 --- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi +++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi @@ -455,11 +455,7 @@ class MaxPooling2D(Layer[tf.Tensor, tf.Tensor]): MaxPool2D = MaxPooling2D class GlobalAveragePooling2D(Layer[tf.Tensor, tf.Tensor]): - def __init__( - self, - data_format: None | Literal["channels_last", "channels_first"] = None, - keepdims: bool = False, - ) -> None: ... + def __init__(self, data_format: None | Literal["channels_last", "channels_first"] = None, keepdims: bool = False) -> None: ... GlobalAvgPool2D = GlobalAveragePooling2D