basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

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

@@ -0,0 +1,116 @@
+"""
+@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
+import tensorflow.core.framework.tensor_pb2
+import tensorflow.core.framework.tensor_shape_pb2
+import tensorflow.core.framework.types_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class CompressedComponentMetadata(google.protobuf.message.Message):
+    """This file contains protocol buffers for working with tf.data Datasets.
+
+    Metadata describing a compressed component of a dataset element.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DTYPE_FIELD_NUMBER: builtins.int
+    TENSOR_SHAPE_FIELD_NUMBER: builtins.int
+    UNCOMPRESSED_BYTES_FIELD_NUMBER: builtins.int
+    dtype: tensorflow.core.framework.types_pb2.DataType.ValueType
+    """The dtype of the component tensor."""
+    @property
+    def tensor_shape(self) -> tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto:
+        """The shape of the component tensor."""
+
+    @property
+    def uncompressed_bytes(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """The amount of uncompressed tensor data.
+        - For string tensors, there is an element for each string indicating the
+        size of the string.
+        - For all other tensors, there is a single element indicating the size of
+        the tensor.
+        """
+
+    def __init__(
+        self,
+        *,
+        dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ...,
+        tensor_shape: tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto | None = ...,
+        uncompressed_bytes: collections.abc.Iterable[builtins.int] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["tensor_shape", b"tensor_shape"]) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "dtype", b"dtype", "tensor_shape", b"tensor_shape", "uncompressed_bytes", b"uncompressed_bytes"
+        ],
+    ) -> None: ...
+
+global___CompressedComponentMetadata = CompressedComponentMetadata
+
+@typing.final
+class CompressedElement(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_FIELD_NUMBER: builtins.int
+    COMPONENT_METADATA_FIELD_NUMBER: builtins.int
+    VERSION_FIELD_NUMBER: builtins.int
+    data: builtins.bytes
+    """Compressed tensor bytes for all components of the element."""
+    version: builtins.int
+    """Version of the CompressedElement. CompressedElements may be stored on disk
+    and read back by later versions of code, so we store a version number to
+    help readers understand which version they are reading. When you add a new
+    field to this proto, you need to increment kCompressedElementVersion in
+    tensorflow/core/data/compression_utils.cc.
+    """
+    @property
+    def component_metadata(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CompressedComponentMetadata]:
+        """Metadata for the components of the element."""
+
+    def __init__(
+        self,
+        *,
+        data: builtins.bytes | None = ...,
+        component_metadata: collections.abc.Iterable[global___CompressedComponentMetadata] | None = ...,
+        version: builtins.int | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self, field_name: typing.Literal["component_metadata", b"component_metadata", "data", b"data", "version", b"version"]
+    ) -> None: ...
+
+global___CompressedElement = CompressedElement
+
+@typing.final
+class UncompressedElement(google.protobuf.message.Message):
+    """An uncompressed dataset element."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    COMPONENTS_FIELD_NUMBER: builtins.int
+    @property
+    def components(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
+        tensorflow.core.framework.tensor_pb2.TensorProto
+    ]: ...
+    def __init__(
+        self, *, components: collections.abc.Iterable[tensorflow.core.framework.tensor_pb2.TensorProto] | None = ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["components", b"components"]) -> None: ...
+
+global___UncompressedElement = UncompressedElement

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

@@ -0,0 +1,140 @@
+"""
+@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 InterconnectLink(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DEVICE_ID_FIELD_NUMBER: builtins.int
+    TYPE_FIELD_NUMBER: builtins.int
+    STRENGTH_FIELD_NUMBER: builtins.int
+    device_id: builtins.int
+    type: builtins.str
+    strength: builtins.int
+    def __init__(
+        self, *, device_id: builtins.int | None = ..., type: builtins.str | None = ..., strength: builtins.int | None = ...
+    ) -> None: ...
+    def ClearField(
+        self, field_name: typing.Literal["device_id", b"device_id", "strength", b"strength", "type", b"type"]
+    ) -> None: ...
+
+global___InterconnectLink = InterconnectLink
+
+@typing.final
+class LocalLinks(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LINK_FIELD_NUMBER: builtins.int
+    @property
+    def link(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___InterconnectLink]: ...
+    def __init__(self, *, link: collections.abc.Iterable[global___InterconnectLink] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["link", b"link"]) -> None: ...
+
+global___LocalLinks = LocalLinks
+
+@typing.final
+class DeviceLocality(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUS_ID_FIELD_NUMBER: builtins.int
+    NUMA_NODE_FIELD_NUMBER: builtins.int
+    LINKS_FIELD_NUMBER: builtins.int
+    bus_id: builtins.int
+    """Optional bus locality of device.  Default value of 0 means
+    no specific locality.  Specific localities are indexed from 1.
+    """
+    numa_node: builtins.int
+    """Optional NUMA locality of device."""
+    @property
+    def links(self) -> global___LocalLinks:
+        """Optional local interconnect links to other devices."""
+
+    def __init__(
+        self, *, bus_id: builtins.int | None = ..., numa_node: builtins.int | None = ..., links: global___LocalLinks | None = ...
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["links", b"links"]) -> builtins.bool: ...
+    def ClearField(
+        self, field_name: typing.Literal["bus_id", b"bus_id", "links", b"links", "numa_node", b"numa_node"]
+    ) -> None: ...
+
+global___DeviceLocality = DeviceLocality
+
+@typing.final
+class DeviceAttributes(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    DEVICE_TYPE_FIELD_NUMBER: builtins.int
+    MEMORY_LIMIT_FIELD_NUMBER: builtins.int
+    LOCALITY_FIELD_NUMBER: builtins.int
+    INCARNATION_FIELD_NUMBER: builtins.int
+    PHYSICAL_DEVICE_DESC_FIELD_NUMBER: builtins.int
+    XLA_GLOBAL_ID_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Fully specified name of the device within a cluster."""
+    device_type: builtins.str
+    """String representation of device_type."""
+    memory_limit: builtins.int
+    """Memory capacity of device in bytes."""
+    incarnation: builtins.int
+    """A device is assigned a global unique number each time it is
+    initialized. "incarnation" should never be 0.
+    """
+    physical_device_desc: builtins.str
+    """String representation of the physical device that this device maps to."""
+    xla_global_id: builtins.int
+    """A physical device ID for use in XLA DeviceAssignments, unique across
+    clients in a multi-client setup. Set to -1 if unavailable, non-negative
+    otherwise.
+    """
+    @property
+    def locality(self) -> global___DeviceLocality:
+        """Platform-specific data about device that may be useful
+        for supporting efficient data transfers.
+        """
+
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        device_type: builtins.str | None = ...,
+        memory_limit: builtins.int | None = ...,
+        locality: global___DeviceLocality | None = ...,
+        incarnation: builtins.int | None = ...,
+        physical_device_desc: builtins.str | None = ...,
+        xla_global_id: builtins.int | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["locality", b"locality"]) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "device_type",
+            b"device_type",
+            "incarnation",
+            b"incarnation",
+            "locality",
+            b"locality",
+            "memory_limit",
+            b"memory_limit",
+            "name",
+            b"name",
+            "physical_device_desc",
+            b"physical_device_desc",
+            "xla_global_id",
+            b"xla_global_id",
+        ],
+    ) -> None: ...
+
+global___DeviceAttributes = DeviceAttributes