Skip to content

Commit

Permalink
Move some layers to keras.layers.preprocessing
Browse files Browse the repository at this point in the history
  • Loading branch information
hoel-bagard committed Jan 28, 2024
1 parent 22c88bc commit 1b20240
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from _typeshed import Incomplete
from collections.abc import Callable, Iterable, Mapping, Sequence
from typing import Any, Generic, Literal, TypeVar, overload
from typing_extensions import Self, TypeAlias
from typing_extensions import Self, TypeAlias, override

import tensorflow as tf
from tensorflow import Tensor, Variable, VariableAggregation, VariableSynchronization, _TensorCompatible
Expand Down Expand Up @@ -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: ...
Expand Down

0 comments on commit 1b20240

Please sign in to comment.