basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

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

@@ -0,0 +1,199 @@
+"""
+@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 DebugTensorWatch(google.protobuf.message.Message):
+    """Option for watching a node in TensorFlow Debugger (tfdbg)."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NODE_NAME_FIELD_NUMBER: builtins.int
+    OUTPUT_SLOT_FIELD_NUMBER: builtins.int
+    DEBUG_OPS_FIELD_NUMBER: builtins.int
+    DEBUG_URLS_FIELD_NUMBER: builtins.int
+    TOLERATE_DEBUG_OP_CREATION_FAILURES_FIELD_NUMBER: builtins.int
+    node_name: builtins.str
+    """Name of the node to watch.
+    Use "*" for wildcard. But note: currently, regex is not supported in
+    general.
+    """
+    output_slot: builtins.int
+    """Output slot to watch.
+    The semantics of output_slot == -1 is that all outputs of the node
+    will be watched (i.e., a wildcard).
+    Other negative values of output_slot are invalid and will lead to
+    errors currently.
+    """
+    tolerate_debug_op_creation_failures: builtins.bool
+    """Do not error out if debug op creation fails (e.g., due to dtype
+    incompatibility). Instead, just log the failure.
+    """
+    @property
+    def debug_ops(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Name(s) of the debugging op(s).
+        One or more than one probes on a tensor.
+        e.g., {"DebugIdentity", "DebugNanCount"}
+        """
+
+    @property
+    def debug_urls(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """URL(s) for debug targets(s).
+
+        Supported URL formats are:
+          - file:///foo/tfdbg_dump: Writes out Event content to file
+            /foo/tfdbg_dump.  Assumes all directories can be created if they don't
+            already exist.
+          - grpc://localhost:11011: Sends an RPC request to an EventListener
+            service running at localhost:11011 with the event.
+          - memcbk:///event_key: Routes tensors to clients using the
+            callback registered with the DebugCallbackRegistry for event_key.
+
+        Each debug op listed in debug_ops will publish its output tensor (debug
+        signal) to all URLs in debug_urls.
+
+        N.B. Session::Run() supports concurrent invocations of the same inputs
+        (feed keys), outputs and target nodes. If such concurrent invocations
+        are to be debugged, the callers of Session::Run() must use distinct
+        debug_urls to make sure that the streamed or dumped events do not overlap
+        among the invocations.
+        TODO(cais): More visible documentation of this in g3docs.
+        """
+
+    def __init__(
+        self,
+        *,
+        node_name: builtins.str | None = ...,
+        output_slot: builtins.int | None = ...,
+        debug_ops: collections.abc.Iterable[builtins.str] | None = ...,
+        debug_urls: collections.abc.Iterable[builtins.str] | None = ...,
+        tolerate_debug_op_creation_failures: builtins.bool | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "debug_ops",
+            b"debug_ops",
+            "debug_urls",
+            b"debug_urls",
+            "node_name",
+            b"node_name",
+            "output_slot",
+            b"output_slot",
+            "tolerate_debug_op_creation_failures",
+            b"tolerate_debug_op_creation_failures",
+        ],
+    ) -> None: ...
+
+global___DebugTensorWatch = DebugTensorWatch
+
+@typing.final
+class DebugOptions(google.protobuf.message.Message):
+    """Options for initializing DebuggerState in TensorFlow Debugger (tfdbg)."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DEBUG_TENSOR_WATCH_OPTS_FIELD_NUMBER: builtins.int
+    GLOBAL_STEP_FIELD_NUMBER: builtins.int
+    RESET_DISK_BYTE_USAGE_FIELD_NUMBER: builtins.int
+    global_step: builtins.int
+    """Caller-specified global step count.
+    Note that this is distinct from the session run count and the executor
+    step count.
+    """
+    reset_disk_byte_usage: builtins.bool
+    """Whether the total disk usage of tfdbg is to be reset to zero
+    in this Session.run call. This is used by wrappers and hooks
+    such as the local CLI ones to indicate that the dumped tensors
+    are cleaned up from the disk after each Session.run.
+    """
+    @property
+    def debug_tensor_watch_opts(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DebugTensorWatch]:
+        """Debugging options"""
+
+    def __init__(
+        self,
+        *,
+        debug_tensor_watch_opts: collections.abc.Iterable[global___DebugTensorWatch] | None = ...,
+        global_step: builtins.int | None = ...,
+        reset_disk_byte_usage: builtins.bool | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "debug_tensor_watch_opts",
+            b"debug_tensor_watch_opts",
+            "global_step",
+            b"global_step",
+            "reset_disk_byte_usage",
+            b"reset_disk_byte_usage",
+        ],
+    ) -> None: ...
+
+global___DebugOptions = DebugOptions
+
+@typing.final
+class DebuggedSourceFile(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    HOST_FIELD_NUMBER: builtins.int
+    FILE_PATH_FIELD_NUMBER: builtins.int
+    LAST_MODIFIED_FIELD_NUMBER: builtins.int
+    BYTES_FIELD_NUMBER: builtins.int
+    LINES_FIELD_NUMBER: builtins.int
+    host: builtins.str
+    """The host name on which a source code file is located."""
+    file_path: builtins.str
+    """Path to the source code file."""
+    last_modified: builtins.int
+    """The timestamp at which the source code file is last modified."""
+    bytes: builtins.int
+    """Byte size of the file."""
+    @property
+    def lines(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Line-by-line content of the source code file."""
+
+    def __init__(
+        self,
+        *,
+        host: builtins.str | None = ...,
+        file_path: builtins.str | None = ...,
+        last_modified: builtins.int | None = ...,
+        bytes: builtins.int | None = ...,
+        lines: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "bytes", b"bytes", "file_path", b"file_path", "host", b"host", "last_modified", b"last_modified", "lines", b"lines"
+        ],
+    ) -> None: ...
+
+global___DebuggedSourceFile = DebuggedSourceFile
+
+@typing.final
+class DebuggedSourceFiles(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOURCE_FILES_FIELD_NUMBER: builtins.int
+    @property
+    def source_files(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DebuggedSourceFile]:
+        """A collection of source code files."""
+
+    def __init__(self, *, source_files: collections.abc.Iterable[global___DebuggedSourceFile] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["source_files", b"source_files"]) -> None: ...
+
+global___DebuggedSourceFiles = DebuggedSourceFiles

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

@@ -0,0 +1,124 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2020 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================
+"""
+
+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 TaskDeviceFilters(google.protobuf.message.Message):
+    """This file contains protos to be used when defining a TensorFlow
+    cluster.
+
+    Configure device filters for remote tasks in the cluster. When associated
+    with a ClusterDef in setting up the cluster, a remote task will ignore all
+    devices which do not match any of its filters. Device filters must be
+    configured at the cluster startup, and cannot be updated once the cluster is
+    up and running.
+
+    EXAMPLES
+    --------
+
+    A two-job cluster with the following ClusterDef:
+
+     Cluster:
+       job { name: 'worker' tasks { key: 0 value: 'worker1:2222' }
+                            tasks { key: 1 value: 'worker2:2222' } }
+       job { name: 'ps'     tasks { key: 0 value: 'ps0:2222' }
+                            tasks { key: 1 value: 'ps1:2222' } }
+
+    Set device filters to isolate worker tasks:
+
+     ClusterDeviceFilters:
+       job { name: 'worker' tasks { key: 0
+                                    value: device_filter '/job:ps'
+                                           device_filter '/job:worker/task:0' }
+                            tasks { key: 1
+                                    value: device_filter '/job:ps'
+                                           device_filter '/job:worker/task:1' } }
+
+    Defines the device filters for a remote task.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DEVICE_FILTERS_FIELD_NUMBER: builtins.int
+    @property
+    def device_filters(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(self, *, device_filters: collections.abc.Iterable[builtins.str] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["device_filters", b"device_filters"]) -> None: ...
+
+global___TaskDeviceFilters = TaskDeviceFilters
+
+@typing.final
+class JobDeviceFilters(google.protobuf.message.Message):
+    """Defines the device filters for tasks in a job."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class TasksEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.int
+        @property
+        def value(self) -> global___TaskDeviceFilters: ...
+        def __init__(self, *, key: builtins.int | None = ..., value: global___TaskDeviceFilters | None = ...) -> None: ...
+        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    NAME_FIELD_NUMBER: builtins.int
+    TASKS_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """The name of this job."""
+    @property
+    def tasks(self) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___TaskDeviceFilters]:
+        """Mapping from task ID to task device filters."""
+
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        tasks: collections.abc.Mapping[builtins.int, global___TaskDeviceFilters] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["name", b"name", "tasks", b"tasks"]) -> None: ...
+
+global___JobDeviceFilters = JobDeviceFilters
+
+@typing.final
+class ClusterDeviceFilters(google.protobuf.message.Message):
+    """Defines the device filters for jobs in a cluster."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    JOBS_FIELD_NUMBER: builtins.int
+    @property
+    def jobs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JobDeviceFilters]: ...
+    def __init__(self, *, jobs: collections.abc.Iterable[global___JobDeviceFilters] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["jobs", b"jobs"]) -> None: ...
+
+global___ClusterDeviceFilters = ClusterDeviceFilters

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

@@ -0,0 +1,154 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2017 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================
+"""
+
+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 DeviceProperties(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class EnvironmentEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    TYPE_FIELD_NUMBER: builtins.int
+    VENDOR_FIELD_NUMBER: builtins.int
+    MODEL_FIELD_NUMBER: builtins.int
+    FREQUENCY_FIELD_NUMBER: builtins.int
+    NUM_CORES_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_FIELD_NUMBER: builtins.int
+    NUM_REGISTERS_FIELD_NUMBER: builtins.int
+    L1_CACHE_SIZE_FIELD_NUMBER: builtins.int
+    L2_CACHE_SIZE_FIELD_NUMBER: builtins.int
+    L3_CACHE_SIZE_FIELD_NUMBER: builtins.int
+    SHARED_MEMORY_SIZE_PER_MULTIPROCESSOR_FIELD_NUMBER: builtins.int
+    MEMORY_SIZE_FIELD_NUMBER: builtins.int
+    BANDWIDTH_FIELD_NUMBER: builtins.int
+    type: builtins.str
+    """Device type (CPU, GPU, ...)"""
+    vendor: builtins.str
+    """Vendor (Intel, nvidia, ...)"""
+    model: builtins.str
+    """Model (Haswell, K40, ...)"""
+    frequency: builtins.int
+    """Core Frequency in Mhz"""
+    num_cores: builtins.int
+    """Number of cores"""
+    num_registers: builtins.int
+    """Number of registers per core."""
+    l1_cache_size: builtins.int
+    """L1 cache size in bytes"""
+    l2_cache_size: builtins.int
+    """L2 cache size in bytes"""
+    l3_cache_size: builtins.int
+    """L3 cache size in bytes"""
+    shared_memory_size_per_multiprocessor: builtins.int
+    """Shared memory size per multiprocessor in bytes. This field is
+    applicable to GPUs only.
+    """
+    memory_size: builtins.int
+    """Memory size in bytes"""
+    bandwidth: builtins.int
+    """Memory bandwidth in KB/s"""
+    @property
+    def environment(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """Version of the tools and libraries used with this device (e.g. gcc 4.9,
+        cudnn 5.1)
+        """
+
+    def __init__(
+        self,
+        *,
+        type: builtins.str | None = ...,
+        vendor: builtins.str | None = ...,
+        model: builtins.str | None = ...,
+        frequency: builtins.int | None = ...,
+        num_cores: builtins.int | None = ...,
+        environment: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        num_registers: builtins.int | None = ...,
+        l1_cache_size: builtins.int | None = ...,
+        l2_cache_size: builtins.int | None = ...,
+        l3_cache_size: builtins.int | None = ...,
+        shared_memory_size_per_multiprocessor: builtins.int | None = ...,
+        memory_size: builtins.int | None = ...,
+        bandwidth: builtins.int | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "bandwidth",
+            b"bandwidth",
+            "environment",
+            b"environment",
+            "frequency",
+            b"frequency",
+            "l1_cache_size",
+            b"l1_cache_size",
+            "l2_cache_size",
+            b"l2_cache_size",
+            "l3_cache_size",
+            b"l3_cache_size",
+            "memory_size",
+            b"memory_size",
+            "model",
+            b"model",
+            "num_cores",
+            b"num_cores",
+            "num_registers",
+            b"num_registers",
+            "shared_memory_size_per_multiprocessor",
+            b"shared_memory_size_per_multiprocessor",
+            "type",
+            b"type",
+            "vendor",
+            b"vendor",
+        ],
+    ) -> None: ...
+
+global___DeviceProperties = DeviceProperties
+
+@typing.final
+class NamedDevice(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    PROPERTIES_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    @property
+    def properties(self) -> global___DeviceProperties: ...
+    def __init__(self, *, name: builtins.str | None = ..., properties: global___DeviceProperties | None = ...) -> None: ...
+    def HasField(self, field_name: typing.Literal["properties", b"properties"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["name", b"name", "properties", b"properties"]) -> None: ...
+
+global___NamedDevice = NamedDevice

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

@@ -0,0 +1,33 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Add a dummy package name. Having no package, like
+core/lib/core/error_codes.proto, or having tensorflow.error, like
+tsl/protobuf/error_codes.proto, results in name collision errors in generated
+code for some users that use JS through J2CL.
+"""
+
+import google.protobuf.descriptor
+from tensorflow.tsl.protobuf.error_codes_pb2 import (
+    ABORTED as ABORTED,
+    ALREADY_EXISTS as ALREADY_EXISTS,
+    CANCELLED as CANCELLED,
+    DATA_LOSS as DATA_LOSS,
+    DEADLINE_EXCEEDED as DEADLINE_EXCEEDED,
+    DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_ as DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_,
+    FAILED_PRECONDITION as FAILED_PRECONDITION,
+    INTERNAL as INTERNAL,
+    INVALID_ARGUMENT as INVALID_ARGUMENT,
+    NOT_FOUND as NOT_FOUND,
+    OK as OK,
+    OUT_OF_RANGE as OUT_OF_RANGE,
+    PERMISSION_DENIED as PERMISSION_DENIED,
+    RESOURCE_EXHAUSTED as RESOURCE_EXHAUSTED,
+    UNAUTHENTICATED as UNAUTHENTICATED,
+    UNAVAILABLE as UNAVAILABLE,
+    UNIMPLEMENTED as UNIMPLEMENTED,
+    UNKNOWN as UNKNOWN,
+    Code as Code,
+)
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

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

@@ -0,0 +1,76 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.message
+import tensorflow.core.framework.versions_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class FingerprintDef(google.protobuf.message.Message):
+    """Protocol buffer representing a SavedModel Fingerprint.
+
+    If there are multiple MetaGraphDefs in the SavedModel, the FingerprintDef
+    corresponds to the first one.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SAVED_MODEL_CHECKSUM_FIELD_NUMBER: builtins.int
+    GRAPH_DEF_PROGRAM_HASH_FIELD_NUMBER: builtins.int
+    SIGNATURE_DEF_HASH_FIELD_NUMBER: builtins.int
+    SAVED_OBJECT_GRAPH_HASH_FIELD_NUMBER: builtins.int
+    CHECKPOINT_HASH_FIELD_NUMBER: builtins.int
+    VERSION_FIELD_NUMBER: builtins.int
+    saved_model_checksum: builtins.int
+    """Hash of the saved_model.pb, referred to as a "checksum"."""
+    graph_def_program_hash: builtins.int
+    """Hash of regularized graph_def."""
+    signature_def_hash: builtins.int
+    """Hash of the regularized (sorted) SignatureDefs."""
+    saved_object_graph_hash: builtins.int
+    """Hash of the regularized SavedObjectGraph."""
+    checkpoint_hash: builtins.int
+    """Hash of the checkpoint."""
+    @property
+    def version(self) -> tensorflow.core.framework.versions_pb2.VersionDef:
+        """Version specification of the fingerprint.
+        TODO(b/290068219): add USM version when GA
+        """
+
+    def __init__(
+        self,
+        *,
+        saved_model_checksum: builtins.int | None = ...,
+        graph_def_program_hash: builtins.int | None = ...,
+        signature_def_hash: builtins.int | None = ...,
+        saved_object_graph_hash: builtins.int | None = ...,
+        checkpoint_hash: builtins.int | None = ...,
+        version: tensorflow.core.framework.versions_pb2.VersionDef | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["version", b"version"]) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "checkpoint_hash",
+            b"checkpoint_hash",
+            "graph_def_program_hash",
+            b"graph_def_program_hash",
+            "saved_model_checksum",
+            b"saved_model_checksum",
+            "saved_object_graph_hash",
+            b"saved_object_graph_hash",
+            "signature_def_hash",
+            b"signature_def_hash",
+            "version",
+            b"version",
+        ],
+    ) -> None: ...
+
+global___FingerprintDef = FingerprintDef