basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/framework/tensor_shape_pb2.pyi

@@ -0,0 +1,74 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Protocol buffer representing the shape of tensors."""
+
+import builtins
+import collections.abc
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class TensorShapeProto(google.protobuf.message.Message):
+    """Dimensions of a tensor."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class Dim(google.protobuf.message.Message):
+        """One dimension of the tensor."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        SIZE_FIELD_NUMBER: builtins.int
+        NAME_FIELD_NUMBER: builtins.int
+        size: builtins.int
+        """Size of the tensor in that dimension.
+        This value must be >= -1, but values of -1 are reserved for "unknown"
+        shapes (values of -1 mean "unknown" dimension).  Certain wrappers
+        that work with TensorShapeProto may fail at runtime when deserializing
+        a TensorShapeProto containing a dim value of -1.
+        """
+        name: builtins.str
+        """Optional name of the tensor dimension."""
+        def __init__(self, *, size: builtins.int | None = ..., name: builtins.str | None = ...) -> None: ...
+        def ClearField(self, field_name: typing.Literal["name", b"name", "size", b"size"]) -> None: ...
+
+    DIM_FIELD_NUMBER: builtins.int
+    UNKNOWN_RANK_FIELD_NUMBER: builtins.int
+    unknown_rank: builtins.bool
+    """If true, the number of dimensions in the shape is unknown.
+
+    If true, "dim.size()" must be 0.
+    """
+    @property
+    def dim(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TensorShapeProto.Dim]:
+        """Dimensions of the tensor, such as {"input", 30}, {"output", 40}
+        for a 30 x 40 2D tensor.  If an entry has size -1, this
+        corresponds to a dimension of unknown size. The names are
+        optional.
+
+        The order of entries in "dim" matters: It indicates the layout of the
+        values in the tensor in-memory representation.
+
+        The first entry in "dim" is the outermost dimension used to layout the
+        values, the last entry is the innermost dimension.  This matches the
+        in-memory layout of RowMajor Eigen tensors.
+
+        If "dim.size()" > 0, "unknown_rank" must be false.
+        """
+
+    def __init__(
+        self,
+        *,
+        dim: collections.abc.Iterable[global___TensorShapeProto.Dim] | None = ...,
+        unknown_rank: builtins.bool | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["dim", b"dim", "unknown_rank", b"unknown_rank"]) -> None: ...
+
+global___TensorShapeProto = TensorShapeProto

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/framework/tensor_slice_pb2.pyi

@@ -0,0 +1,56 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Protocol buffer representing slices of a tensor"""
+
+import builtins
+import collections.abc
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class TensorSliceProto(google.protobuf.message.Message):
+    """Can only be interpreted if you know the corresponding TensorShape."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class Extent(google.protobuf.message.Message):
+        """Extent of the slice in one dimension.
+        Either both or no attributes must be set.  When no attribute is set
+        means: All data in that dimension.
+        """
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        START_FIELD_NUMBER: builtins.int
+        LENGTH_FIELD_NUMBER: builtins.int
+        start: builtins.int
+        """Start index of the slice, starting at 0."""
+        length: builtins.int
+        def __init__(self, *, start: builtins.int | None = ..., length: builtins.int | None = ...) -> None: ...
+        def HasField(self, field_name: typing.Literal["has_length", b"has_length", "length", b"length"]) -> builtins.bool: ...
+        def ClearField(
+            self, field_name: typing.Literal["has_length", b"has_length", "length", b"length", "start", b"start"]
+        ) -> None: ...
+        def WhichOneof(self, oneof_group: typing.Literal["has_length", b"has_length"]) -> typing.Literal["length"] | None: ...
+
+    EXTENT_FIELD_NUMBER: builtins.int
+    @property
+    def extent(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TensorSliceProto.Extent]:
+        """Extent of the slice in all tensor dimensions.
+
+        Must have one entry for each of the dimension of the tensor that this
+        slice belongs to.  The order of sizes is the same as the order of
+        dimensions in the TensorShape.
+        """
+
+    def __init__(self, *, extent: collections.abc.Iterable[global___TensorSliceProto.Extent] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["extent", b"extent"]) -> None: ...
+
+global___TensorSliceProto = TensorSliceProto

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/framework/types_pb2.pyi

@@ -0,0 +1,220 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import sys
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _DataType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _DataTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DataType.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DT_INVALID: _DataType.ValueType  # 0
+    """Not a legal value for DataType.  Used to indicate a DataType field
+    has not been set.
+    """
+    DT_FLOAT: _DataType.ValueType  # 1
+    """Data types that all computation devices are expected to be
+    capable to support.
+    """
+    DT_DOUBLE: _DataType.ValueType  # 2
+    DT_INT32: _DataType.ValueType  # 3
+    DT_UINT8: _DataType.ValueType  # 4
+    DT_INT16: _DataType.ValueType  # 5
+    DT_INT8: _DataType.ValueType  # 6
+    DT_STRING: _DataType.ValueType  # 7
+    DT_COMPLEX64: _DataType.ValueType  # 8
+    """Single-precision complex"""
+    DT_INT64: _DataType.ValueType  # 9
+    DT_BOOL: _DataType.ValueType  # 10
+    DT_QINT8: _DataType.ValueType  # 11
+    """Quantized int8"""
+    DT_QUINT8: _DataType.ValueType  # 12
+    """Quantized uint8"""
+    DT_QINT32: _DataType.ValueType  # 13
+    """Quantized int32"""
+    DT_BFLOAT16: _DataType.ValueType  # 14
+    """Float32 truncated to 16 bits."""
+    DT_QINT16: _DataType.ValueType  # 15
+    """Quantized int16"""
+    DT_QUINT16: _DataType.ValueType  # 16
+    """Quantized uint16"""
+    DT_UINT16: _DataType.ValueType  # 17
+    DT_COMPLEX128: _DataType.ValueType  # 18
+    """Double-precision complex"""
+    DT_HALF: _DataType.ValueType  # 19
+    DT_RESOURCE: _DataType.ValueType  # 20
+    DT_VARIANT: _DataType.ValueType  # 21
+    """Arbitrary C++ data types"""
+    DT_UINT32: _DataType.ValueType  # 22
+    DT_UINT64: _DataType.ValueType  # 23
+    DT_FLOAT8_E5M2: _DataType.ValueType  # 24
+    """5 exponent bits, 2 mantissa bits."""
+    DT_FLOAT8_E4M3FN: _DataType.ValueType  # 25
+    """4 exponent bits, 3 mantissa bits, finite-only, with"""
+    DT_INT4: _DataType.ValueType  # 29
+    """2 NaNs (0bS1111111).
+    TODO - b/299182407: Leaving room for remaining float8 types.
+    DT_FLOAT8_E4M3FNUZ = 26;
+    DT_FLOAT8_E4M3B11FNUZ = 27;
+    DT_FLOAT8_E5M2FNUZ = 28;
+    """
+    DT_UINT4: _DataType.ValueType  # 30
+    DT_FLOAT_REF: _DataType.ValueType  # 101
+    """Do not use!  These are only for TF1's obsolete reference Variables.
+    Every enum above should have a corresponding value below (verified by
+    types_test).
+    """
+    DT_DOUBLE_REF: _DataType.ValueType  # 102
+    DT_INT32_REF: _DataType.ValueType  # 103
+    DT_UINT8_REF: _DataType.ValueType  # 104
+    DT_INT16_REF: _DataType.ValueType  # 105
+    DT_INT8_REF: _DataType.ValueType  # 106
+    DT_STRING_REF: _DataType.ValueType  # 107
+    DT_COMPLEX64_REF: _DataType.ValueType  # 108
+    DT_INT64_REF: _DataType.ValueType  # 109
+    DT_BOOL_REF: _DataType.ValueType  # 110
+    DT_QINT8_REF: _DataType.ValueType  # 111
+    DT_QUINT8_REF: _DataType.ValueType  # 112
+    DT_QINT32_REF: _DataType.ValueType  # 113
+    DT_BFLOAT16_REF: _DataType.ValueType  # 114
+    DT_QINT16_REF: _DataType.ValueType  # 115
+    DT_QUINT16_REF: _DataType.ValueType  # 116
+    DT_UINT16_REF: _DataType.ValueType  # 117
+    DT_COMPLEX128_REF: _DataType.ValueType  # 118
+    DT_HALF_REF: _DataType.ValueType  # 119
+    DT_RESOURCE_REF: _DataType.ValueType  # 120
+    DT_VARIANT_REF: _DataType.ValueType  # 121
+    DT_UINT32_REF: _DataType.ValueType  # 122
+    DT_UINT64_REF: _DataType.ValueType  # 123
+    DT_FLOAT8_E5M2_REF: _DataType.ValueType  # 124
+    DT_FLOAT8_E4M3FN_REF: _DataType.ValueType  # 125
+    DT_INT4_REF: _DataType.ValueType  # 129
+    """TODO - b/299182407: Leaving room for remaining float8 types.
+    DT_FLOAT8_E4M3FNUZ_REF = 126;
+    DT_FLOAT8_E4M3B11FNUZ_REF = 127;
+    DT_FLOAT8_E5M2FNUZ_REF = 128;
+    """
+    DT_UINT4_REF: _DataType.ValueType  # 130
+
+class DataType(_DataType, metaclass=_DataTypeEnumTypeWrapper):
+    """(== suppress_warning documentation-presence ==)
+    LINT.IfChange
+    """
+
+DT_INVALID: DataType.ValueType  # 0
+"""Not a legal value for DataType.  Used to indicate a DataType field
+has not been set.
+"""
+DT_FLOAT: DataType.ValueType  # 1
+"""Data types that all computation devices are expected to be
+capable to support.
+"""
+DT_DOUBLE: DataType.ValueType  # 2
+DT_INT32: DataType.ValueType  # 3
+DT_UINT8: DataType.ValueType  # 4
+DT_INT16: DataType.ValueType  # 5
+DT_INT8: DataType.ValueType  # 6
+DT_STRING: DataType.ValueType  # 7
+DT_COMPLEX64: DataType.ValueType  # 8
+"""Single-precision complex"""
+DT_INT64: DataType.ValueType  # 9
+DT_BOOL: DataType.ValueType  # 10
+DT_QINT8: DataType.ValueType  # 11
+"""Quantized int8"""
+DT_QUINT8: DataType.ValueType  # 12
+"""Quantized uint8"""
+DT_QINT32: DataType.ValueType  # 13
+"""Quantized int32"""
+DT_BFLOAT16: DataType.ValueType  # 14
+"""Float32 truncated to 16 bits."""
+DT_QINT16: DataType.ValueType  # 15
+"""Quantized int16"""
+DT_QUINT16: DataType.ValueType  # 16
+"""Quantized uint16"""
+DT_UINT16: DataType.ValueType  # 17
+DT_COMPLEX128: DataType.ValueType  # 18
+"""Double-precision complex"""
+DT_HALF: DataType.ValueType  # 19
+DT_RESOURCE: DataType.ValueType  # 20
+DT_VARIANT: DataType.ValueType  # 21
+"""Arbitrary C++ data types"""
+DT_UINT32: DataType.ValueType  # 22
+DT_UINT64: DataType.ValueType  # 23
+DT_FLOAT8_E5M2: DataType.ValueType  # 24
+"""5 exponent bits, 2 mantissa bits."""
+DT_FLOAT8_E4M3FN: DataType.ValueType  # 25
+"""4 exponent bits, 3 mantissa bits, finite-only, with"""
+DT_INT4: DataType.ValueType  # 29
+"""2 NaNs (0bS1111111).
+TODO - b/299182407: Leaving room for remaining float8 types.
+DT_FLOAT8_E4M3FNUZ = 26;
+DT_FLOAT8_E4M3B11FNUZ = 27;
+DT_FLOAT8_E5M2FNUZ = 28;
+"""
+DT_UINT4: DataType.ValueType  # 30
+DT_FLOAT_REF: DataType.ValueType  # 101
+"""Do not use!  These are only for TF1's obsolete reference Variables.
+Every enum above should have a corresponding value below (verified by
+types_test).
+"""
+DT_DOUBLE_REF: DataType.ValueType  # 102
+DT_INT32_REF: DataType.ValueType  # 103
+DT_UINT8_REF: DataType.ValueType  # 104
+DT_INT16_REF: DataType.ValueType  # 105
+DT_INT8_REF: DataType.ValueType  # 106
+DT_STRING_REF: DataType.ValueType  # 107
+DT_COMPLEX64_REF: DataType.ValueType  # 108
+DT_INT64_REF: DataType.ValueType  # 109
+DT_BOOL_REF: DataType.ValueType  # 110
+DT_QINT8_REF: DataType.ValueType  # 111
+DT_QUINT8_REF: DataType.ValueType  # 112
+DT_QINT32_REF: DataType.ValueType  # 113
+DT_BFLOAT16_REF: DataType.ValueType  # 114
+DT_QINT16_REF: DataType.ValueType  # 115
+DT_QUINT16_REF: DataType.ValueType  # 116
+DT_UINT16_REF: DataType.ValueType  # 117
+DT_COMPLEX128_REF: DataType.ValueType  # 118
+DT_HALF_REF: DataType.ValueType  # 119
+DT_RESOURCE_REF: DataType.ValueType  # 120
+DT_VARIANT_REF: DataType.ValueType  # 121
+DT_UINT32_REF: DataType.ValueType  # 122
+DT_UINT64_REF: DataType.ValueType  # 123
+DT_FLOAT8_E5M2_REF: DataType.ValueType  # 124
+DT_FLOAT8_E4M3FN_REF: DataType.ValueType  # 125
+DT_INT4_REF: DataType.ValueType  # 129
+"""TODO - b/299182407: Leaving room for remaining float8 types.
+DT_FLOAT8_E4M3FNUZ_REF = 126;
+DT_FLOAT8_E4M3B11FNUZ_REF = 127;
+DT_FLOAT8_E5M2FNUZ_REF = 128;
+"""
+DT_UINT4_REF: DataType.ValueType  # 130
+global___DataType = DataType
+
+@typing.final
+class SerializedDType(google.protobuf.message.Message):
+    """Represents a serialized tf.dtypes.Dtype"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATATYPE_FIELD_NUMBER: builtins.int
+    datatype: global___DataType.ValueType
+    def __init__(self, *, datatype: global___DataType.ValueType | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["datatype", b"datatype"]) -> None: ...
+
+global___SerializedDType = SerializedDType

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/framework/variable_pb2.pyi

@@ -0,0 +1,229 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import sys
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _VariableSynchronization:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _VariableSynchronizationEnumTypeWrapper(
+    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VariableSynchronization.ValueType], builtins.type
+):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    VARIABLE_SYNCHRONIZATION_AUTO: _VariableSynchronization.ValueType  # 0
+    """`AUTO`: Indicates that the synchronization will be determined by the
+    current `DistributionStrategy` (eg. With `MirroredStrategy` this would be
+    `ON_WRITE`).
+    """
+    VARIABLE_SYNCHRONIZATION_NONE: _VariableSynchronization.ValueType  # 1
+    """`NONE`: Indicates that there will only be one copy of the variable, so
+    there is no need to sync.
+    """
+    VARIABLE_SYNCHRONIZATION_ON_WRITE: _VariableSynchronization.ValueType  # 2
+    """`ON_WRITE`: Indicates that the variable will be updated across devices
+    every time it is written.
+    """
+    VARIABLE_SYNCHRONIZATION_ON_READ: _VariableSynchronization.ValueType  # 3
+    """`ON_READ`: Indicates that the variable will be aggregated across devices
+    when it is read (eg. when checkpointing or when evaluating an op that uses
+    the variable).
+    """
+
+class VariableSynchronization(_VariableSynchronization, metaclass=_VariableSynchronizationEnumTypeWrapper):
+    """Indicates when a distributed variable will be synced."""
+
+VARIABLE_SYNCHRONIZATION_AUTO: VariableSynchronization.ValueType  # 0
+"""`AUTO`: Indicates that the synchronization will be determined by the
+current `DistributionStrategy` (eg. With `MirroredStrategy` this would be
+`ON_WRITE`).
+"""
+VARIABLE_SYNCHRONIZATION_NONE: VariableSynchronization.ValueType  # 1
+"""`NONE`: Indicates that there will only be one copy of the variable, so
+there is no need to sync.
+"""
+VARIABLE_SYNCHRONIZATION_ON_WRITE: VariableSynchronization.ValueType  # 2
+"""`ON_WRITE`: Indicates that the variable will be updated across devices
+every time it is written.
+"""
+VARIABLE_SYNCHRONIZATION_ON_READ: VariableSynchronization.ValueType  # 3
+"""`ON_READ`: Indicates that the variable will be aggregated across devices
+when it is read (eg. when checkpointing or when evaluating an op that uses
+the variable).
+"""
+global___VariableSynchronization = VariableSynchronization
+
+class _VariableAggregation:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _VariableAggregationEnumTypeWrapper(
+    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VariableAggregation.ValueType], builtins.type
+):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    VARIABLE_AGGREGATION_NONE: _VariableAggregation.ValueType  # 0
+    """`NONE`: This is the default, giving an error if you use a
+    variable-update operation with multiple replicas.
+    """
+    VARIABLE_AGGREGATION_SUM: _VariableAggregation.ValueType  # 1
+    """`SUM`: Add the updates across replicas."""
+    VARIABLE_AGGREGATION_MEAN: _VariableAggregation.ValueType  # 2
+    """`MEAN`: Take the arithmetic mean ("average") of the updates across
+    replicas.
+    """
+    VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA: _VariableAggregation.ValueType  # 3
+    """`ONLY_FIRST_REPLICA`: This is for when every replica is performing the same
+    update, but we only want to perform the update once. Used, e.g., for the
+    global step counter.
+    """
+
+class VariableAggregation(_VariableAggregation, metaclass=_VariableAggregationEnumTypeWrapper):
+    """Indicates how a distributed variable will be aggregated."""
+
+VARIABLE_AGGREGATION_NONE: VariableAggregation.ValueType  # 0
+"""`NONE`: This is the default, giving an error if you use a
+variable-update operation with multiple replicas.
+"""
+VARIABLE_AGGREGATION_SUM: VariableAggregation.ValueType  # 1
+"""`SUM`: Add the updates across replicas."""
+VARIABLE_AGGREGATION_MEAN: VariableAggregation.ValueType  # 2
+"""`MEAN`: Take the arithmetic mean ("average") of the updates across
+replicas.
+"""
+VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA: VariableAggregation.ValueType  # 3
+"""`ONLY_FIRST_REPLICA`: This is for when every replica is performing the same
+update, but we only want to perform the update once. Used, e.g., for the
+global step counter.
+"""
+global___VariableAggregation = VariableAggregation
+
+@typing.final
+class VariableDef(google.protobuf.message.Message):
+    """Protocol buffer representing a Variable."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VARIABLE_NAME_FIELD_NUMBER: builtins.int
+    INITIAL_VALUE_NAME_FIELD_NUMBER: builtins.int
+    INITIALIZER_NAME_FIELD_NUMBER: builtins.int
+    SNAPSHOT_NAME_FIELD_NUMBER: builtins.int
+    SAVE_SLICE_INFO_DEF_FIELD_NUMBER: builtins.int
+    IS_RESOURCE_FIELD_NUMBER: builtins.int
+    TRAINABLE_FIELD_NUMBER: builtins.int
+    SYNCHRONIZATION_FIELD_NUMBER: builtins.int
+    AGGREGATION_FIELD_NUMBER: builtins.int
+    variable_name: builtins.str
+    """Name of the variable tensor."""
+    initial_value_name: builtins.str
+    """Name of the tensor holding the variable's initial value."""
+    initializer_name: builtins.str
+    """Name of the initializer op."""
+    snapshot_name: builtins.str
+    """Name of the snapshot tensor."""
+    is_resource: builtins.bool
+    """Whether to represent this as a ResourceVariable."""
+    trainable: builtins.bool
+    """Whether this variable should be trained."""
+    synchronization: global___VariableSynchronization.ValueType
+    """Indicates when a distributed variable will be synced."""
+    aggregation: global___VariableAggregation.ValueType
+    """Indicates how a distributed variable will be aggregated."""
+    @property
+    def save_slice_info_def(self) -> global___SaveSliceInfoDef:
+        """Support for saving variables as slices of a larger variable."""
+
+    def __init__(
+        self,
+        *,
+        variable_name: builtins.str | None = ...,
+        initial_value_name: builtins.str | None = ...,
+        initializer_name: builtins.str | None = ...,
+        snapshot_name: builtins.str | None = ...,
+        save_slice_info_def: global___SaveSliceInfoDef | None = ...,
+        is_resource: builtins.bool | None = ...,
+        trainable: builtins.bool | None = ...,
+        synchronization: global___VariableSynchronization.ValueType | None = ...,
+        aggregation: global___VariableAggregation.ValueType | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["save_slice_info_def", b"save_slice_info_def"]) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "aggregation",
+            b"aggregation",
+            "initial_value_name",
+            b"initial_value_name",
+            "initializer_name",
+            b"initializer_name",
+            "is_resource",
+            b"is_resource",
+            "save_slice_info_def",
+            b"save_slice_info_def",
+            "snapshot_name",
+            b"snapshot_name",
+            "synchronization",
+            b"synchronization",
+            "trainable",
+            b"trainable",
+            "variable_name",
+            b"variable_name",
+        ],
+    ) -> None: ...
+
+global___VariableDef = VariableDef
+
+@typing.final
+class SaveSliceInfoDef(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FULL_NAME_FIELD_NUMBER: builtins.int
+    FULL_SHAPE_FIELD_NUMBER: builtins.int
+    VAR_OFFSET_FIELD_NUMBER: builtins.int
+    VAR_SHAPE_FIELD_NUMBER: builtins.int
+    full_name: builtins.str
+    """Name of the full variable of which this is a slice."""
+    @property
+    def full_shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """Shape of the full variable."""
+
+    @property
+    def var_offset(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """Offset of this variable into the full variable."""
+
+    @property
+    def var_shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """Shape of this variable."""
+
+    def __init__(
+        self,
+        *,
+        full_name: builtins.str | None = ...,
+        full_shape: collections.abc.Iterable[builtins.int] | None = ...,
+        var_offset: collections.abc.Iterable[builtins.int] | None = ...,
+        var_shape: collections.abc.Iterable[builtins.int] | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "full_name", b"full_name", "full_shape", b"full_shape", "var_offset", b"var_offset", "var_shape", b"var_shape"
+        ],
+    ) -> None: ...
+
+global___SaveSliceInfoDef = SaveSliceInfoDef

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/framework/versions_pb2.pyi

@@ -0,0 +1,57 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class VersionDef(google.protobuf.message.Message):
+    """Version information for a piece of serialized data
+
+    There are different types of versions for each type of data
+    (GraphDef, etc.), but they all have the same common shape
+    described here.
+
+    Each consumer has "consumer" and "min_producer" versions (specified
+    elsewhere).  A consumer is allowed to consume this data if
+
+      producer >= min_producer
+      consumer >= min_consumer
+      consumer not in bad_consumers
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PRODUCER_FIELD_NUMBER: builtins.int
+    MIN_CONSUMER_FIELD_NUMBER: builtins.int
+    BAD_CONSUMERS_FIELD_NUMBER: builtins.int
+    producer: builtins.int
+    """The version of the code that produced this data."""
+    min_consumer: builtins.int
+    """Any consumer below this version is not allowed to consume this data."""
+    @property
+    def bad_consumers(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """Specific consumer versions which are disallowed (e.g. due to bugs)."""
+
+    def __init__(
+        self,
+        *,
+        producer: builtins.int | None = ...,
+        min_consumer: builtins.int | None = ...,
+        bad_consumers: collections.abc.Iterable[builtins.int] | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal["bad_consumers", b"bad_consumers", "min_consumer", b"min_consumer", "producer", b"producer"],
+    ) -> None: ...
+
+global___VersionDef = VersionDef

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/protobuf/bfc_memory_map_pb2.pyi

@@ -0,0 +1,15 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import google.protobuf.descriptor
+from tensorflow.compiler.xla.tsl.protobuf.bfc_memory_map_pb2 import (
+    BinSummary as BinSummary,
+    MemAllocatorStats as MemAllocatorStats,
+    MemChunk as MemChunk,
+    MemoryDump as MemoryDump,
+    SnapShot as SnapShot,
+)
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor