basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/util/saved_tensor_slice_pb2.pyi

@@ -0,0 +1,171 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+A checkpoint file is an sstable. The value for each record is a serialized
+SavedTensorSlices message (defined below).
+
+Each checkpoint file has a record with the empty key (""), which corresponds
+to a SavedTensorSlices message that contains a "meta", that serves as a
+table of contents on all the tensor slices saved in this file. Since the key
+is "", it's always the first record in each file.
+
+Each of the rest of the records in a checkpoint stores the raw data of a
+particular tensor slice, in SavedSlice format. The corresponding key is an
+ordered code that encodes the name of the tensor and the slice
+information. The name is also stored in the SaveSlice message for ease of
+debugging and manual examination.
+"""
+
+import builtins
+import collections.abc
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import tensorflow.core.framework.tensor_pb2
+import tensorflow.core.framework.tensor_shape_pb2
+import tensorflow.core.framework.tensor_slice_pb2
+import tensorflow.core.framework.types_pb2
+import tensorflow.core.framework.versions_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class SavedSliceMeta(google.protobuf.message.Message):
+    """Metadata describing the set of slices of the same tensor saved in a
+    checkpoint file.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    SHAPE_FIELD_NUMBER: builtins.int
+    TYPE_FIELD_NUMBER: builtins.int
+    SLICE_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Name of the tensor."""
+    type: tensorflow.core.framework.types_pb2.DataType.ValueType
+    """Type of the tensor"""
+    @property
+    def shape(self) -> tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto:
+        """Shape of the tensor"""
+
+    @property
+    def slice(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
+        tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto
+    ]:
+        """Explicit list of slices saved in the checkpoint file."""
+
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        shape: tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto | None = ...,
+        type: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ...,
+        slice: collections.abc.Iterable[tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["shape", b"shape"]) -> builtins.bool: ...
+    def ClearField(
+        self, field_name: typing.Literal["name", b"name", "shape", b"shape", "slice", b"slice", "type", b"type"]
+    ) -> None: ...
+
+global___SavedSliceMeta = SavedSliceMeta
+
+@typing.final
+class SavedTensorSliceMeta(google.protobuf.message.Message):
+    """Metadata describing the set of tensor slices saved in a checkpoint file.
+    It is always stored at the beginning of each checkpoint file.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TENSOR_FIELD_NUMBER: builtins.int
+    VERSIONS_FIELD_NUMBER: builtins.int
+    @property
+    def tensor(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SavedSliceMeta]:
+        """Each SavedSliceMeta describes the slices for one tensor."""
+
+    @property
+    def versions(self) -> tensorflow.core.framework.versions_pb2.VersionDef:
+        """Compatibility version of this checkpoint.  See core/public/version.h
+        for version history.
+        """
+
+    def __init__(
+        self,
+        *,
+        tensor: collections.abc.Iterable[global___SavedSliceMeta] | None = ...,
+        versions: tensorflow.core.framework.versions_pb2.VersionDef | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["versions", b"versions"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["tensor", b"tensor", "versions", b"versions"]) -> None: ...
+
+global___SavedTensorSliceMeta = SavedTensorSliceMeta
+
+@typing.final
+class SavedSlice(google.protobuf.message.Message):
+    """Saved tensor slice: it stores the name of the tensors, the slice, and the
+    raw data.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    SLICE_FIELD_NUMBER: builtins.int
+    DATA_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Name of the tensor that this slice belongs to. This must be identical to
+    the name used to encode the key for this record.
+    """
+    @property
+    def slice(self) -> tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto:
+        """Extent of the slice.  Must have one entry for each of the dimension of the
+        tensor that this slice belongs to.
+        """
+
+    @property
+    def data(self) -> tensorflow.core.framework.tensor_pb2.TensorProto:
+        """The raw data of the slice is stored as a TensorProto. Only raw data are
+        stored (we don't fill in fields such as dtype or tensor_shape).
+        """
+
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        slice: tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto | None = ...,
+        data: tensorflow.core.framework.tensor_pb2.TensorProto | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["data", b"data", "slice", b"slice"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["data", b"data", "name", b"name", "slice", b"slice"]) -> None: ...
+
+global___SavedSlice = SavedSlice
+
+@typing.final
+class SavedTensorSlices(google.protobuf.message.Message):
+    """Each record in a v3 checkpoint file is a serialized SavedTensorSlices
+    message.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    META_FIELD_NUMBER: builtins.int
+    DATA_FIELD_NUMBER: builtins.int
+    @property
+    def meta(self) -> global___SavedTensorSliceMeta:
+        """This is only present at the first item of each checkpoint file and serves
+        as a table of contents, listing all the tensor slices saved in this file.
+        """
+
+    @property
+    def data(self) -> global___SavedSlice:
+        """This exists in all but the first item of each checkpoint file."""
+
+    def __init__(self, *, meta: global___SavedTensorSliceMeta | None = ..., data: global___SavedSlice | None = ...) -> None: ...
+    def HasField(self, field_name: typing.Literal["data", b"data", "meta", b"meta"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["data", b"data", "meta", b"meta"]) -> None: ...
+
+global___SavedTensorSlices = SavedTensorSlices

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/util/test_log_pb2.pyi

@@ -0,0 +1,24 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Protocol messages for describing the results of benchmarks and unit tests."""
+
+import google.protobuf.descriptor
+from tensorflow.compiler.xla.tsl.protobuf.test_log_pb2 import (
+    AvailableDeviceInfo as AvailableDeviceInfo,
+    BenchmarkEntries as BenchmarkEntries,
+    BenchmarkEntry as BenchmarkEntry,
+    BuildConfiguration as BuildConfiguration,
+    CommitId as CommitId,
+    CPUInfo as CPUInfo,
+    EntryValue as EntryValue,
+    GPUInfo as GPUInfo,
+    MachineConfiguration as MachineConfiguration,
+    MemoryInfo as MemoryInfo,
+    MetricEntry as MetricEntry,
+    PlatformInfo as PlatformInfo,
+    RunConfiguration as RunConfiguration,
+    TestResults as TestResults,
+)
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/data/__init__.pyi

@@ -0,0 +1,272 @@
+from _typeshed import Incomplete
+from abc import ABC, abstractmethod
+from collections.abc import Callable, Collection, Iterator as _Iterator, Sequence
+from typing import Any, Generic, TypeVar, overload
+from typing_extensions import Self, Unpack
+
+import numpy as np
+import tensorflow as tf
+from tensorflow import TypeSpec
+from tensorflow._aliases import ContainerGeneric, ScalarTensorCompatible, TensorCompatible
+from tensorflow.data import experimental as experimental
+from tensorflow.data.experimental import AUTOTUNE as AUTOTUNE
+from tensorflow.dtypes import DType
+from tensorflow.io import _CompressionTypes
+from tensorflow.python.trackable.base import Trackable
+
+_T1_co = TypeVar("_T1_co", covariant=True)
+_T2 = TypeVar("_T2")
+_T3 = TypeVar("_T3")
+
+class Iterator(_Iterator[_T1_co], Trackable, ABC):
+    @property
+    @abstractmethod
+    def element_spec(self) -> ContainerGeneric[TypeSpec[Any]]: ...
+    @abstractmethod
+    def get_next(self) -> _T1_co: ...
+    @abstractmethod
+    def get_next_as_optional(self) -> tf.experimental.Optional[_T1_co]: ...
+
+class Dataset(ABC, Generic[_T1_co]):
+    def apply(self, transformation_func: Callable[[Dataset[_T1_co]], Dataset[_T2]]) -> Dataset[_T2]: ...
+    def as_numpy_iterator(self) -> Iterator[np.ndarray[Any, Any]]: ...
+    def batch(
+        self,
+        batch_size: ScalarTensorCompatible,
+        drop_remainder: bool = False,
+        num_parallel_calls: int | None = None,
+        deterministic: bool | None = None,
+        name: str | None = None,
+    ) -> Dataset[_T1_co]: ...
+    def bucket_by_sequence_length(
+        self,
+        element_length_func: Callable[[_T1_co], ScalarTensorCompatible],
+        bucket_boundaries: Sequence[int],
+        bucket_batch_sizes: Sequence[int],
+        padded_shapes: ContainerGeneric[tf.TensorShape | TensorCompatible] | None = None,
+        padding_values: ContainerGeneric[ScalarTensorCompatible] | None = None,
+        pad_to_bucket_boundary: bool = False,
+        no_padding: bool = False,
+        drop_remainder: bool = False,
+        name: str | None = None,
+    ) -> Dataset[_T1_co]: ...
+    def cache(self, filename: str = "", name: str | None = None) -> Dataset[_T1_co]: ...
+    def cardinality(self) -> int: ...
+    @staticmethod
+    def choose_from_datasets(
+        datasets: Sequence[Dataset[_T2]], choice_dataset: Dataset[tf.Tensor], stop_on_empty_dataset: bool = True
+    ) -> Dataset[_T2]: ...
+    def concatenate(self, dataset: Dataset[_T1_co], name: str | None = None) -> Dataset[_T1_co]: ...
+    @staticmethod
+    def counter(
+        start: ScalarTensorCompatible = 0, step: ScalarTensorCompatible = 1, dtype: DType = ..., name: str | None = None
+    ) -> Dataset[tf.Tensor]: ...
+    @property
+    @abstractmethod
+    def element_spec(self) -> ContainerGeneric[TypeSpec[Any]]: ...
+    def enumerate(self, start: ScalarTensorCompatible = 0, name: str | None = None) -> Dataset[tuple[int, _T1_co]]: ...
+    def filter(self, predicate: Callable[[_T1_co], bool | tf.Tensor], name: str | None = None) -> Dataset[_T1_co]: ...
+    def flat_map(self, map_func: Callable[[_T1_co], Dataset[_T2]], name: str | None = None) -> Dataset[_T2]: ...
+    # PEP 646 can be used here for a more precise type when better supported.
+    @staticmethod
+    def from_generator(
+        generator: Callable[..., _T2],
+        output_types: ContainerGeneric[DType] | None = None,
+        output_shapes: ContainerGeneric[tf.TensorShape | Sequence[int | None]] | None = None,
+        args: tuple[object, ...] | None = None,
+        output_signature: ContainerGeneric[TypeSpec[Any]] | None = None,
+        name: str | None = None,
+    ) -> Dataset[_T2]: ...
+    @staticmethod
+    def from_tensors(tensors: Any, name: str | None = None) -> Dataset[Any]: ...
+    @staticmethod
+    def from_tensor_slices(tensors: TensorCompatible, name: str | None = None) -> Dataset[Any]: ...
+    def get_single_element(self, name: str | None = None) -> _T1_co: ...
+    def group_by_window(
+        self,
+        key_func: Callable[[_T1_co], tf.Tensor],
+        reduce_func: Callable[[tf.Tensor, Dataset[_T1_co]], Dataset[_T2]],
+        window_size: ScalarTensorCompatible | None = None,
+        window_size_func: Callable[[tf.Tensor], tf.Tensor] | None = None,
+        name: str | None = None,
+    ) -> Dataset[_T2]: ...
+    def ignore_errors(self, log_warning: bool = False, name: str | None = None) -> Dataset[_T1_co]: ...
+    def interleave(
+        self,
+        map_func: Callable[[_T1_co], Dataset[_T2]],
+        cycle_length: int | None = None,
+        block_length: int | None = None,
+        num_parallel_calls: int | None = None,
+        deterministic: bool | None = None,
+        name: str | None = None,
+    ) -> Dataset[_T2]: ...
+    def __iter__(self) -> Iterator[_T1_co]: ...
+    @staticmethod
+    def list_files(
+        file_pattern: str | Sequence[str] | TensorCompatible,
+        shuffle: bool | None = None,
+        seed: int | None = None,
+        name: str | None = None,
+    ) -> Dataset[str]: ...
+    @staticmethod
+    def load(
+        path: str,
+        element_spec: ContainerGeneric[tf.TypeSpec[Any]] | None = None,
+        compression: _CompressionTypes = None,
+        reader_func: Callable[[Dataset[Dataset[Any]]], Dataset[Any]] | None = None,
+        wait: bool = False,
+    ) -> Dataset[Any]: ...
+    # PEP 646 could be used here for a more precise type when better supported.
+    def map(
+        self,
+        map_func: Callable[..., _T2],
+        num_parallel_calls: int | None = None,
+        deterministic: bool | None = None,
+        synchronous: bool | None = None,
+        use_unbounded_threadpool: bool = False,
+        name: str | None = None,
+    ) -> Dataset[_T2]: ...
+    def options(self) -> Options: ...
+    def padded_batch(
+        self,
+        batch_size: ScalarTensorCompatible,
+        padded_shapes: ContainerGeneric[tf.TensorShape | TensorCompatible] | None = None,
+        padding_values: ContainerGeneric[ScalarTensorCompatible] | None = None,
+        drop_remainder: bool = False,
+        name: str | None = None,
+    ) -> Dataset[_T1_co]: ...
+    def prefetch(self, buffer_size: ScalarTensorCompatible, name: str | None = None) -> Dataset[_T1_co]: ...
+    def ragged_batch(
+        self,
+        batch_size: ScalarTensorCompatible,
+        drop_remainder: bool = False,
+        row_splits_dtype: DType = ...,
+        name: str | None = None,
+    ) -> Dataset[tf.RaggedTensor]: ...
+    @staticmethod
+    def random(
+        seed: int | None = None, rerandomize_each_iteration: bool | None = None, name: str | None = None
+    ) -> Dataset[tf.Tensor]: ...
+    @staticmethod
+    @overload
+    def range(stop: ScalarTensorCompatible, /, output_type: DType = ..., name: str | None = None) -> Dataset[tf.Tensor]: ...
+    @staticmethod
+    @overload
+    def range(
+        start: ScalarTensorCompatible,
+        stop: ScalarTensorCompatible,
+        step: ScalarTensorCompatible = 1,
+        /,
+        output_type: DType = ...,
+        name: str | None = None,
+    ) -> Dataset[tf.Tensor]: ...
+    def rebatch(
+        self, batch_size: ScalarTensorCompatible, drop_remainder: bool = False, name: str | None = None
+    ) -> Dataset[_T1_co]: ...
+    def reduce(self, initial_state: _T2, reduce_func: Callable[[_T2, _T1_co], _T2], name: str | None = None) -> _T2: ...
+    def rejection_resample(
+        self,
+        class_func: Callable[[_T1_co], ScalarTensorCompatible],
+        target_dist: TensorCompatible,
+        initial_dist: TensorCompatible | None = None,
+        seed: int | None = None,
+        name: str | None = None,
+    ) -> Dataset[_T1_co]: ...
+    def repeat(self, count: ScalarTensorCompatible | None = None, name: str | None = None) -> Dataset[_T1_co]: ...
+    @staticmethod
+    def sample_from_datasets(
+        datasets: Sequence[Dataset[_T1_co]],
+        weights: TensorCompatible | None = None,
+        seed: int | None = None,
+        stop_on_empty_dataset: bool = False,
+        rerandomize_each_iteration: bool | None = None,
+    ) -> Dataset[_T1_co]: ...
+    # Incomplete as tf.train.CheckpointOptions not yet covered.
+    def save(
+        self,
+        path: str,
+        compression: _CompressionTypes = None,
+        shard_func: Callable[[_T1_co], int] | None = None,
+        checkpoint_args=None,
+    ) -> None: ...
+    def scan(
+        self, initial_state: _T2, scan_func: Callable[[_T2, _T1_co], tuple[_T2, _T3]], name: str | None = None
+    ) -> Dataset[_T3]: ...
+    def shard(
+        self, num_shards: ScalarTensorCompatible, index: ScalarTensorCompatible, name: str | None = None
+    ) -> Dataset[_T1_co]: ...
+    def shuffle(
+        self,
+        buffer_size: ScalarTensorCompatible,
+        seed: int | None = None,
+        reshuffle_each_iteration: bool = True,
+        name: str | None = None,
+    ) -> Dataset[_T1_co]: ...
+    def skip(self, count: ScalarTensorCompatible, name: str | None = None) -> Dataset[_T1_co]: ...
+    def snapshot(
+        self,
+        path: str,
+        compression: _CompressionTypes = "AUTO",
+        reader_func: Callable[[Dataset[Dataset[_T1_co]]], Dataset[_T1_co]] | None = None,
+        shard_func: Callable[[_T1_co], ScalarTensorCompatible] | None = None,
+        name: str | None = None,
+    ) -> Dataset[_T1_co]: ...
+    def sparse_batch(
+        self, batch_size: ScalarTensorCompatible, row_shape: tf.TensorShape | TensorCompatible, name: str | None = None
+    ) -> Dataset[tf.SparseTensor]: ...
+    def take(self, count: ScalarTensorCompatible, name: str | None = None) -> Dataset[_T1_co]: ...
+    def take_while(self, predicate: Callable[[_T1_co], ScalarTensorCompatible], name: str | None = None) -> Dataset[_T1_co]: ...
+    def unbatch(self, name: str | None = None) -> Dataset[_T1_co]: ...
+    def unique(self, name: str | None = None) -> Dataset[_T1_co]: ...
+    def window(
+        self,
+        size: ScalarTensorCompatible,
+        shift: ScalarTensorCompatible | None = None,
+        stride: ScalarTensorCompatible = 1,
+        drop_remainder: bool = False,
+        name: str | None = None,
+    ) -> Dataset[Dataset[_T1_co]]: ...
+    def with_options(self, options: Options, name: str | None = None) -> Dataset[_T1_co]: ...
+    @overload
+    @staticmethod
+    def zip(
+        *args: Dataset[Any] | Collection[Dataset[Any]] | ContainerGeneric[Dataset[Any]], name: str | None = None
+    ) -> Dataset[tuple[Any, ...]]: ...
+    @overload
+    @staticmethod
+    def zip(*args: Unpack[tuple[Dataset[_T2], Dataset[_T3]]], name: str | None = None) -> Dataset[tuple[_T2, _T3]]: ...
+    @overload
+    @staticmethod
+    def zip(
+        *, datasets: tuple[Dataset[_T2], Dataset[_T3]] | None = None, name: str | None = None
+    ) -> Dataset[tuple[_T2, _T3]]: ...
+    def __len__(self) -> int: ...
+    def __nonzero__(self) -> bool: ...
+    def __getattr__(self, name: str) -> Incomplete: ...
+
+class Options:
+    autotune: Incomplete
+    deterministic: bool
+    experimental_deterministic: bool
+    experimental_distribute: Incomplete
+    experimental_external_state_policy: Incomplete
+    experimental_optimization: Incomplete
+    experimental_slack: bool
+    experimental_symbolic_checkpoint: bool
+    experimental_threading: Incomplete
+    threading: Incomplete
+    def merge(self, options: Options) -> Self: ...
+
+class TFRecordDataset(Dataset[tf.Tensor]):
+    def __init__(
+        self,
+        filenames: TensorCompatible | Dataset[str],
+        compression_type: _CompressionTypes = None,
+        buffer_size: int | None = None,
+        num_parallel_reads: int | None = None,
+        name: str | None = None,
+    ) -> None: ...
+    @property
+    def element_spec(self) -> tf.TensorSpec: ...
+
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/data/experimental.pyi

@@ -0,0 +1,32 @@
+from collections.abc import Callable, Sequence
+from typing import Final, TypeVar
+
+from tensorflow import Tensor
+from tensorflow._aliases import TensorCompatible
+from tensorflow.data import Dataset
+
+AUTOTUNE: Final = -1
+INFINITE_CARDINALITY: Final = -1
+SHARD_HINT: Final = -1
+UNKNOWN_CARDINALITY: Final = -2
+
+_T1 = TypeVar("_T1")
+_T2 = TypeVar("_T2")
+
+def parallel_interleave(
+    map_func: Callable[[_T1], Dataset[_T2]],
+    cycle_length: int,
+    block_length: int = 1,
+    sloppy: bool | None = False,
+    buffer_output_elements: int | None = None,
+    prefetch_input_elements: int | None = None,
+) -> Callable[[Dataset[_T1]], Dataset[_T2]]: ...
+def enable_debug_mode() -> None: ...
+def cardinality(dataset: Dataset[object]) -> Tensor: ...
+def sample_from_datasets(
+    datasets: Sequence[Dataset[_T1]],
+    weights: TensorCompatible | None = None,
+    seed: int | None = None,
+    stop_on_empty_dataset: bool = False,
+) -> Dataset[_T1]: ...
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/distribute/__init__.pyi

@@ -0,0 +1,3 @@
+from tensorflow.python.distribute.distribute_lib import Strategy as Strategy
+
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/distribute/coordinator.pyi

@@ -0,0 +1,3 @@
+from .experimental.coordinator import RemoteValue as RemoteValue
+
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi

@@ -0,0 +1,11 @@
+from typing import Generic, TypeVar
+
+from tensorflow._aliases import AnyArray
+
+_Value_co = TypeVar("_Value_co", covariant=True)
+
+class RemoteValue(Generic[_Value_co]):
+    def fetch(self) -> AnyArray: ...
+    def get(self) -> _Value_co: ...
+
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/dtypes.pyi

@@ -0,0 +1,57 @@
+from _typeshed import Incomplete
+from abc import ABCMeta
+from builtins import bool as _bool
+from typing import Any
+
+import numpy as np
+from tensorflow._aliases import DTypeLike
+from tensorflow.python.framework.dtypes import HandleData
+
+class _DTypeMeta(ABCMeta): ...
+
+class DType(metaclass=_DTypeMeta):
+    def __init__(self, type_enum: int, handle_data: HandleData | None = None) -> None: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def as_numpy_dtype(self) -> type[np.number[Any]]: ...
+    @property
+    def is_numpy_compatible(self) -> _bool: ...
+    @property
+    def is_bool(self) -> _bool: ...
+    @property
+    def is_floating(self) -> _bool: ...
+    @property
+    def is_integer(self) -> _bool: ...
+    @property
+    def is_quantized(self) -> _bool: ...
+    @property
+    def is_unsigned(self) -> _bool: ...
+    def __getattr__(self, name: str) -> Incomplete: ...
+
+bool: DType
+complex128: DType
+complex64: DType
+bfloat16: DType
+float16: DType
+half: DType
+float32: DType
+float64: DType
+double: DType
+int8: DType
+int16: DType
+int32: DType
+int64: DType
+uint8: DType
+uint16: DType
+uint32: DType
+uint64: DType
+qint8: DType
+qint16: DType
+qint32: DType
+quint8: DType
+quint16: DType
+string: DType
+
+def as_dtype(type_value: DTypeLike) -> DType: ...
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/experimental/__init__.pyi

@@ -0,0 +1,10 @@
+from _typeshed import Incomplete
+from abc import ABC
+from typing import Generic, TypeVar
+
+_T_co = TypeVar("_T_co", covariant=True)
+
+class Optional(ABC, Generic[_T_co]):
+    def __getattr__(self, name: str) -> Incomplete: ...
+
+def __getattr__(name: str): ...  # incomplete module

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/experimental/dtensor.pyi

@@ -0,0 +1,19 @@
+from _typeshed import Incomplete
+
+from tensorflow._aliases import IntArray, IntDataSequence
+
+Layout = Incomplete
+
+class Mesh:
+    def __init__(
+        self,
+        dim_names: list[str],
+        global_device_ids: IntArray | IntDataSequence,
+        local_device_ids: list[int],
+        local_devices: list[Incomplete | str],
+        mesh_name: str = "",
+        global_devices: list[Incomplete | str] | None = None,
+        use_xla_spmd: bool = False,
+    ) -> None: ...
+
+def __getattr__(name: str): ...  # incomplete module