Move some layers to keras.layers.preprocessing

python · Jan 28, 2024 · 52e0d83 · 52e0d83
1 parent 22c88bc
commit 52e0d83
Showing 1 changed file with 8 additions and 8 deletions.
diff --git 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: ...