basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

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

@@ -0,0 +1,391 @@
+"""
+@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.attr_value_pb2
+import tensorflow.core.framework.full_type_pb2
+import tensorflow.core.framework.resource_handle_pb2
+import tensorflow.core.framework.types_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class OpDef(google.protobuf.message.Message):
+    """Defines an operation. A NodeDef in a GraphDef specifies an Op by
+    using the "op" field which should match the name of a OpDef.
+    LINT.IfChange
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class ArgDef(google.protobuf.message.Message):
+        """For describing inputs and outputs."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        NAME_FIELD_NUMBER: builtins.int
+        DESCRIPTION_FIELD_NUMBER: builtins.int
+        TYPE_FIELD_NUMBER: builtins.int
+        TYPE_ATTR_FIELD_NUMBER: builtins.int
+        NUMBER_ATTR_FIELD_NUMBER: builtins.int
+        TYPE_LIST_ATTR_FIELD_NUMBER: builtins.int
+        HANDLE_DATA_FIELD_NUMBER: builtins.int
+        IS_REF_FIELD_NUMBER: builtins.int
+        EXPERIMENTAL_FULL_TYPE_FIELD_NUMBER: builtins.int
+        name: builtins.str
+        """Name for the input/output.  Should match the regexp "[a-z][a-z0-9_]*"."""
+        description: builtins.str
+        """Human readable description."""
+        type: tensorflow.core.framework.types_pb2.DataType.ValueType
+        """Describes the type of one or more tensors that are accepted/produced
+        by this input/output arg.  The only legal combinations are:
+        * For a single tensor: either the "type" field is set or the
+          "type_attr" field is set to the name of an attr with type "type".
+        * For a sequence of tensors with the same type: the "number_attr"
+          field will be set to the name of an attr with type "int", and
+          either the "type" or "type_attr" field will be set as for
+          single tensors.
+        * For a sequence of tensors, the "type_list_attr" field will be set
+          to the name of an attr with type "list(type)".
+        """
+        type_attr: builtins.str
+        """if specified, attr must have type "type" """
+        number_attr: builtins.str
+        """if specified, attr must have type "int" """
+        type_list_attr: builtins.str
+        """If specified, attr must have type "list(type)", and none of
+        type, type_attr, and number_attr may be specified.
+        """
+        is_ref: builtins.bool
+        """For inputs: if true, the inputs are required to be refs.
+          By default, inputs can be either refs or non-refs.
+        For outputs: if true, outputs are refs, otherwise they are not.
+        """
+        @property
+        def handle_data(
+            self,
+        ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
+            tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto.DtypeAndShape
+        ]:
+            """The handle data for resource inputs."""
+
+        @property
+        def experimental_full_type(self) -> tensorflow.core.framework.full_type_pb2.FullTypeDef:
+            """Experimental. Full type declaration for this argument.
+            The full type specification combines type, type_attr, type_list_attr,
+            etc. into a unified representation.
+            This declaration may contain non-concrete types (for example,
+            Tensor<TypeVar<'T'>> is a valid type declaration.
+
+            Note: this is a transient field. The long-term aim is to represent the
+            entire OpDef as a single type: a callable. In that context, this field is
+            just the type of a single argument.
+            """
+
+        def __init__(
+            self,
+            *,
+            name: builtins.str | None = ...,
+            description: builtins.str | None = ...,
+            type: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ...,
+            type_attr: builtins.str | None = ...,
+            number_attr: builtins.str | None = ...,
+            type_list_attr: builtins.str | None = ...,
+            handle_data: (
+                collections.abc.Iterable[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto.DtypeAndShape] | None
+            ) = ...,
+            is_ref: builtins.bool | None = ...,
+            experimental_full_type: tensorflow.core.framework.full_type_pb2.FullTypeDef | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["experimental_full_type", b"experimental_full_type"]) -> builtins.bool: ...
+        def ClearField(
+            self,
+            field_name: typing.Literal[
+                "description",
+                b"description",
+                "experimental_full_type",
+                b"experimental_full_type",
+                "handle_data",
+                b"handle_data",
+                "is_ref",
+                b"is_ref",
+                "name",
+                b"name",
+                "number_attr",
+                b"number_attr",
+                "type",
+                b"type",
+                "type_attr",
+                b"type_attr",
+                "type_list_attr",
+                b"type_list_attr",
+            ],
+        ) -> None: ...
+
+    @typing.final
+    class AttrDef(google.protobuf.message.Message):
+        """Description of the graph-construction-time configuration of this
+        Op.  That is to say, this describes the attr fields that will
+        be specified in the NodeDef.
+        """
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        NAME_FIELD_NUMBER: builtins.int
+        TYPE_FIELD_NUMBER: builtins.int
+        DEFAULT_VALUE_FIELD_NUMBER: builtins.int
+        DESCRIPTION_FIELD_NUMBER: builtins.int
+        HAS_MINIMUM_FIELD_NUMBER: builtins.int
+        MINIMUM_FIELD_NUMBER: builtins.int
+        ALLOWED_VALUES_FIELD_NUMBER: builtins.int
+        name: builtins.str
+        """A descriptive name for the argument.  May be used, e.g. by the
+        Python client, as a keyword argument name, and so should match
+        the regexp "[a-z][a-z0-9_]+".
+        """
+        type: builtins.str
+        """One of the type names from attr_value.proto ("string", "list(string)",
+        "int", etc.).
+        """
+        description: builtins.str
+        """Human-readable description."""
+        has_minimum: builtins.bool
+        """--- Constraints ---
+        These constraints are only in effect if specified.  Default is no
+        constraints.
+
+        For type == "int", this is a minimum value.  For "list(___)"
+        types, this is the minimum length.
+        """
+        minimum: builtins.int
+        @property
+        def default_value(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue:
+            """A reasonable default for this attribute if the user does not supply
+            a value.  If not specified, the user must supply a value.
+            """
+
+        @property
+        def allowed_values(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue:
+            """The set of allowed values.  Has type that is the "list" version
+            of the "type" field above (uses the "list" field of AttrValue).
+            If type == "type" or "list(type)" above, then the "type" field
+            of "allowed_values.list" has the set of allowed DataTypes.
+            If type == "string" or "list(string)", then the "s" field of
+            "allowed_values.list" has the set of allowed strings.
+            """
+
+        def __init__(
+            self,
+            *,
+            name: builtins.str | None = ...,
+            type: builtins.str | None = ...,
+            default_value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ...,
+            description: builtins.str | None = ...,
+            has_minimum: builtins.bool | None = ...,
+            minimum: builtins.int | None = ...,
+            allowed_values: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ...,
+        ) -> None: ...
+        def HasField(
+            self, field_name: typing.Literal["allowed_values", b"allowed_values", "default_value", b"default_value"]
+        ) -> builtins.bool: ...
+        def ClearField(
+            self,
+            field_name: typing.Literal[
+                "allowed_values",
+                b"allowed_values",
+                "default_value",
+                b"default_value",
+                "description",
+                b"description",
+                "has_minimum",
+                b"has_minimum",
+                "minimum",
+                b"minimum",
+                "name",
+                b"name",
+                "type",
+                b"type",
+            ],
+        ) -> None: ...
+
+    NAME_FIELD_NUMBER: builtins.int
+    INPUT_ARG_FIELD_NUMBER: builtins.int
+    OUTPUT_ARG_FIELD_NUMBER: builtins.int
+    CONTROL_OUTPUT_FIELD_NUMBER: builtins.int
+    ATTR_FIELD_NUMBER: builtins.int
+    DEPRECATION_FIELD_NUMBER: builtins.int
+    SUMMARY_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    IS_COMMUTATIVE_FIELD_NUMBER: builtins.int
+    IS_AGGREGATE_FIELD_NUMBER: builtins.int
+    IS_STATEFUL_FIELD_NUMBER: builtins.int
+    ALLOWS_UNINITIALIZED_INPUT_FIELD_NUMBER: builtins.int
+    IS_DISTRIBUTED_COMMUNICATION_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Op names starting with an underscore are reserved for internal use.
+    Names should be CamelCase and match the regexp "[A-Z][a-zA-Z0-9>_]*".
+    """
+    summary: builtins.str
+    """One-line human-readable description of what the Op does."""
+    description: builtins.str
+    """Additional, longer human-readable description of what the Op does."""
+    is_commutative: builtins.bool
+    """-------------------------------------------------------------------------
+    Which optimizations this operation can participate in.
+
+    True if the operation is commutative ("op(a,b) == op(b,a)" for all inputs)
+    """
+    is_aggregate: builtins.bool
+    """If is_aggregate is true, then this operation accepts N >= 2
+    inputs and produces 1 output all of the same type.  Should be
+    associative and commutative, and produce output with the same
+    shape as the input.  The optimizer may replace an aggregate op
+    taking input from multiple devices with a tree of aggregate ops
+    that aggregate locally within each device (and possibly within
+    groups of nearby devices) before communicating.
+    TODO(josh11b): Implement that optimization.
+    for things like add
+    """
+    is_stateful: builtins.bool
+    """-------------------------------------------------------------------------
+    Optimization constraints.
+
+    Ops are marked as stateful if their behavior depends on some state beyond
+    their input tensors (e.g. variable reading op) or if they have
+    a side-effect (e.g. printing or asserting ops). Equivalently, stateless ops
+    must always produce the same output for the same input and have
+    no side-effects.
+
+    By default Ops may be moved between devices.  Stateful ops should
+    either not be moved, or should only be moved if that state can also
+    be moved (e.g. via some sort of save / restore).
+    Stateful ops are guaranteed to never be optimized away by Common
+    Subexpression Elimination (CSE).
+    for things like variables, queue
+    """
+    allows_uninitialized_input: builtins.bool
+    """-------------------------------------------------------------------------
+    Non-standard options.
+
+    By default, all inputs to an Op must be initialized Tensors.  Ops
+    that may initialize tensors for the first time should set this
+    field to true, to allow the Op to take an uninitialized Tensor as
+    input.
+    for Assign, etc.
+    """
+    is_distributed_communication: builtins.bool
+    """Indicates whether the op implementation uses distributed communication.
+    If True, the op is allowed to return errors for network disconnection and
+    trigger TF network failure handling logics.
+    """
+    @property
+    def input_arg(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpDef.ArgDef]:
+        """Description of the input(s)."""
+
+    @property
+    def output_arg(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpDef.ArgDef]:
+        """Description of the output(s)."""
+
+    @property
+    def control_output(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Named control outputs for this operation. Useful only for composite
+        operations (i.e. functions) which want to name different control outputs.
+        """
+
+    @property
+    def attr(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpDef.AttrDef]: ...
+    @property
+    def deprecation(self) -> global___OpDeprecation:
+        """Optional deprecation based on GraphDef versions."""
+
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        input_arg: collections.abc.Iterable[global___OpDef.ArgDef] | None = ...,
+        output_arg: collections.abc.Iterable[global___OpDef.ArgDef] | None = ...,
+        control_output: collections.abc.Iterable[builtins.str] | None = ...,
+        attr: collections.abc.Iterable[global___OpDef.AttrDef] | None = ...,
+        deprecation: global___OpDeprecation | None = ...,
+        summary: builtins.str | None = ...,
+        description: builtins.str | None = ...,
+        is_commutative: builtins.bool | None = ...,
+        is_aggregate: builtins.bool | None = ...,
+        is_stateful: builtins.bool | None = ...,
+        allows_uninitialized_input: builtins.bool | None = ...,
+        is_distributed_communication: builtins.bool | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["deprecation", b"deprecation"]) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "allows_uninitialized_input",
+            b"allows_uninitialized_input",
+            "attr",
+            b"attr",
+            "control_output",
+            b"control_output",
+            "deprecation",
+            b"deprecation",
+            "description",
+            b"description",
+            "input_arg",
+            b"input_arg",
+            "is_aggregate",
+            b"is_aggregate",
+            "is_commutative",
+            b"is_commutative",
+            "is_distributed_communication",
+            b"is_distributed_communication",
+            "is_stateful",
+            b"is_stateful",
+            "name",
+            b"name",
+            "output_arg",
+            b"output_arg",
+            "summary",
+            b"summary",
+        ],
+    ) -> None: ...
+
+global___OpDef = OpDef
+
+@typing.final
+class OpDeprecation(google.protobuf.message.Message):
+    """Information about version-dependent deprecation of an op"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VERSION_FIELD_NUMBER: builtins.int
+    EXPLANATION_FIELD_NUMBER: builtins.int
+    version: builtins.int
+    """First GraphDef version at which the op is disallowed."""
+    explanation: builtins.str
+    """Explanation of why it was deprecated and what to use instead."""
+    def __init__(self, *, version: builtins.int | None = ..., explanation: builtins.str | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["explanation", b"explanation", "version", b"version"]) -> None: ...
+
+global___OpDeprecation = OpDeprecation
+
+@typing.final
+class OpList(google.protobuf.message.Message):
+    """A collection of OpDefs"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OP_FIELD_NUMBER: builtins.int
+    @property
+    def op(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpDef]: ...
+    def __init__(self, *, op: collections.abc.Iterable[global___OpDef] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["op", b"op"]) -> None: ...
+
+global___OpList = OpList

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

@@ -0,0 +1,166 @@
+"""
+@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
+import tensorflow.core.framework.graph_pb2
+import tensorflow.core.framework.types_pb2
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class OptimizedFunctionGraph(google.protobuf.message.Message):
+    """Optimized function graph after instantiation-related graph optimization
+    passes (up till before graph partitioning). The first half of the proto is
+    representing a GraphDef and the rest of the fields are extra information from
+    graph optimizations.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _OptimizationSource:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _OptimizationSourceEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizedFunctionGraph._OptimizationSource.ValueType],
+        builtins.type,
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        SOURCE_UNSPECIFIED: OptimizedFunctionGraph._OptimizationSource.ValueType  # 0
+        AOT: OptimizedFunctionGraph._OptimizationSource.ValueType  # 1
+        JIT: OptimizedFunctionGraph._OptimizationSource.ValueType  # 2
+
+    class OptimizationSource(_OptimizationSource, metaclass=_OptimizationSourceEnumTypeWrapper):
+        """Enum for distinguishing the origin where the proto is created.
+
+        AOT: proto is created in ahead-of-time environment, which can be different
+        from the environment where the graph is actually executed.
+
+        JIT: proto is created in just-in-time execution, which has the same
+        environment as the one the graph is actually executed.
+        """
+
+    SOURCE_UNSPECIFIED: OptimizedFunctionGraph.OptimizationSource.ValueType  # 0
+    AOT: OptimizedFunctionGraph.OptimizationSource.ValueType  # 1
+    JIT: OptimizedFunctionGraph.OptimizationSource.ValueType  # 2
+
+    @typing.final
+    class NodeNameToControlRetEntry(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: ...
+
+    NAME_FIELD_NUMBER: builtins.int
+    FUNCTION_GRAPH_FIELD_NUMBER: builtins.int
+    NODE_NAME_TO_CONTROL_RET_FIELD_NUMBER: builtins.int
+    RET_TYPES_FIELD_NUMBER: builtins.int
+    NUM_RETURN_NODES_FIELD_NUMBER: builtins.int
+    SOURCE_FIELD_NUMBER: builtins.int
+    OPTIMIZATION_TIME_USECS_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Function name. It can be a human-readable SignatureDef's method name, or a
+    FunctionDef name.
+    """
+    num_return_nodes: builtins.int
+    """Number of return nodes. This is an output of graph preprocessing."""
+    source: global___OptimizedFunctionGraph.OptimizationSource.ValueType
+    """Indicates the source environment where this proto is generated."""
+    optimization_time_usecs: builtins.int
+    """Time (in microseconds) spent on running the graph optimization passes for
+    this function.
+    """
+    @property
+    def function_graph(self) -> tensorflow.core.framework.graph_pb2.GraphDef:
+        """Optimized function graph."""
+
+    @property
+    def node_name_to_control_ret(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """Maps from node name to control ret. This is an output from running TF/XLA
+        bridge.
+        """
+
+    @property
+    def ret_types(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.core.framework.types_pb2.DataType.ValueType]:
+        """Return node types of the function. This is an output of graph
+        preprocessing.
+        """
+
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        function_graph: tensorflow.core.framework.graph_pb2.GraphDef | None = ...,
+        node_name_to_control_ret: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        ret_types: collections.abc.Iterable[tensorflow.core.framework.types_pb2.DataType.ValueType] | None = ...,
+        num_return_nodes: builtins.int | None = ...,
+        source: global___OptimizedFunctionGraph.OptimizationSource.ValueType | None = ...,
+        optimization_time_usecs: builtins.int | None = ...,
+    ) -> None: ...
+    def HasField(
+        self,
+        field_name: typing.Literal[
+            "_optimization_time_usecs",
+            b"_optimization_time_usecs",
+            "_source",
+            b"_source",
+            "function_graph",
+            b"function_graph",
+            "optimization_time_usecs",
+            b"optimization_time_usecs",
+            "source",
+            b"source",
+        ],
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "_optimization_time_usecs",
+            b"_optimization_time_usecs",
+            "_source",
+            b"_source",
+            "function_graph",
+            b"function_graph",
+            "name",
+            b"name",
+            "node_name_to_control_ret",
+            b"node_name_to_control_ret",
+            "num_return_nodes",
+            b"num_return_nodes",
+            "optimization_time_usecs",
+            b"optimization_time_usecs",
+            "ret_types",
+            b"ret_types",
+            "source",
+            b"source",
+        ],
+    ) -> None: ...
+    @typing.overload
+    def WhichOneof(
+        self, oneof_group: typing.Literal["_optimization_time_usecs", b"_optimization_time_usecs"]
+    ) -> typing.Literal["optimization_time_usecs"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_source", b"_source"]) -> typing.Literal["source"] | None: ...
+
+global___OptimizedFunctionGraph = OptimizedFunctionGraph

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

@@ -0,0 +1,52 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class ReaderBaseState(google.protobuf.message.Message):
+    """For serializing and restoring the state of ReaderBase, see
+    reader_base.h for details.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    WORK_STARTED_FIELD_NUMBER: builtins.int
+    WORK_FINISHED_FIELD_NUMBER: builtins.int
+    NUM_RECORDS_PRODUCED_FIELD_NUMBER: builtins.int
+    CURRENT_WORK_FIELD_NUMBER: builtins.int
+    work_started: builtins.int
+    work_finished: builtins.int
+    num_records_produced: builtins.int
+    current_work: builtins.bytes
+    def __init__(
+        self,
+        *,
+        work_started: builtins.int | None = ...,
+        work_finished: builtins.int | None = ...,
+        num_records_produced: builtins.int | None = ...,
+        current_work: builtins.bytes | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "current_work",
+            b"current_work",
+            "num_records_produced",
+            b"num_records_produced",
+            "work_finished",
+            b"work_finished",
+            "work_started",
+            b"work_started",
+        ],
+    ) -> None: ...
+
+global___ReaderBaseState = ReaderBaseState