From 52e0d830a62deac35959ae96c1b65c849f8b0308 Mon Sep 17 00:00:00 2001
From: Hoel Bagard <hoel.bagard@gmail.com>
Date: Sun, 28 Jan 2024 18:34:30 +0900
Subject: [PATCH] Move some layers to keras.layers.preprocessing

---
 .../tensorflow/keras/layers/__init__.pyi         | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
index 6b3f192c1e2e..baac41696d7d 100644
--- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
+++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
@@ -129,7 +129,7 @@ class Dense(Layer[tf.Tensor, tf.Tensor]):
         kernel_constraint: _Constraint = None,
         bias_constraint: _Constraint = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -151,7 +151,7 @@ class BatchNormalization(Layer[tf.Tensor, tf.Tensor]):
         beta_constraint: _Constraint = None,
         gamma_constraint: _Constraint = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -163,7 +163,7 @@ class ReLU(Layer[tf.Tensor, tf.Tensor]):
         negative_slope: float | None = 0.0,
         threshold: float | None = 0.0,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -175,7 +175,7 @@ class Dropout(Layer[tf.Tensor, tf.Tensor]):
         noise_shape: _TensorCompatible | Sequence[int | None] | None = None,
         seed: int | None = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -191,7 +191,7 @@ class Embedding(Layer[tf.Tensor, tf.Tensor]):
         mask_zero: bool = False,
         input_length: int | None = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -216,7 +216,7 @@ class Conv2D(Layer[tf.Tensor, tf.Tensor]):
         kernel_constraint: _Constraint = None,
         bias_constraint: _Constraint = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -240,7 +240,7 @@ class LayerNormalization(Layer[tf.Tensor, tf.Tensor]):
         beta_constraint: _Constraint = None,
         gamma_constraint: _Constraint = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...
@@ -273,7 +273,7 @@ class MultiHeadAttention(Layer[Any, tf.Tensor]):
         kernel_constraint: _Constraint | None = None,
         bias_constraint: _Constraint | None = None,
         trainable: bool = True,
-        dtype: _LayerDtype = None,
+        dtype: _LayerDtype | None = None,
         dynamic: bool = False,
         name: str | None = None,
     ) -> None: ...