basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

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

@@ -0,0 +1,290 @@
+"""
+@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.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 GraphTransferNodeInput(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NODE_ID_FIELD_NUMBER: builtins.int
+    OUTPUT_PORT_FIELD_NUMBER: builtins.int
+    node_id: builtins.int
+    output_port: builtins.int
+    def __init__(self, *, node_id: builtins.int | None = ..., output_port: builtins.int | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["node_id", b"node_id", "output_port", b"output_port"]) -> None: ...
+
+global___GraphTransferNodeInput = GraphTransferNodeInput
+
+@typing.final
+class GraphTransferNodeInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    NODE_ID_FIELD_NUMBER: builtins.int
+    TYPE_NAME_FIELD_NUMBER: builtins.int
+    SOC_OP_ID_FIELD_NUMBER: builtins.int
+    PADDING_ID_FIELD_NUMBER: builtins.int
+    INPUT_COUNT_FIELD_NUMBER: builtins.int
+    OUTPUT_COUNT_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    node_id: builtins.int
+    type_name: builtins.str
+    soc_op_id: builtins.int
+    padding_id: builtins.int
+    input_count: builtins.int
+    output_count: builtins.int
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        node_id: builtins.int | None = ...,
+        type_name: builtins.str | None = ...,
+        soc_op_id: builtins.int | None = ...,
+        padding_id: builtins.int | None = ...,
+        input_count: builtins.int | None = ...,
+        output_count: builtins.int | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "input_count",
+            b"input_count",
+            "name",
+            b"name",
+            "node_id",
+            b"node_id",
+            "output_count",
+            b"output_count",
+            "padding_id",
+            b"padding_id",
+            "soc_op_id",
+            b"soc_op_id",
+            "type_name",
+            b"type_name",
+        ],
+    ) -> None: ...
+
+global___GraphTransferNodeInfo = GraphTransferNodeInfo
+
+@typing.final
+class GraphTransferConstNodeInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    NODE_ID_FIELD_NUMBER: builtins.int
+    SHAPE_FIELD_NUMBER: builtins.int
+    DATA_FIELD_NUMBER: builtins.int
+    DTYPE_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    node_id: builtins.int
+    data: builtins.bytes
+    dtype: tensorflow.core.framework.types_pb2.DataType.ValueType
+    @property
+    def shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        node_id: builtins.int | None = ...,
+        shape: collections.abc.Iterable[builtins.int] | None = ...,
+        data: builtins.bytes | None = ...,
+        dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal["data", b"data", "dtype", b"dtype", "name", b"name", "node_id", b"node_id", "shape", b"shape"],
+    ) -> None: ...
+
+global___GraphTransferConstNodeInfo = GraphTransferConstNodeInfo
+
+@typing.final
+class GraphTransferNodeInputInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NODE_ID_FIELD_NUMBER: builtins.int
+    NODE_INPUT_FIELD_NUMBER: builtins.int
+    node_id: builtins.int
+    @property
+    def node_input(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInput]: ...
+    def __init__(
+        self,
+        *,
+        node_id: builtins.int | None = ...,
+        node_input: collections.abc.Iterable[global___GraphTransferNodeInput] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["node_id", b"node_id", "node_input", b"node_input"]) -> None: ...
+
+global___GraphTransferNodeInputInfo = GraphTransferNodeInputInfo
+
+@typing.final
+class GraphTransferNodeOutputInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NODE_ID_FIELD_NUMBER: builtins.int
+    MAX_BYTE_SIZE_FIELD_NUMBER: builtins.int
+    node_id: builtins.int
+    @property
+    def max_byte_size(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
+    def __init__(
+        self, *, node_id: builtins.int | None = ..., max_byte_size: collections.abc.Iterable[builtins.int] | None = ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["max_byte_size", b"max_byte_size", "node_id", b"node_id"]) -> None: ...
+
+global___GraphTransferNodeOutputInfo = GraphTransferNodeOutputInfo
+
+@typing.final
+class GraphTransferGraphInputNodeInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    SHAPE_FIELD_NUMBER: builtins.int
+    DTYPE_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    dtype: tensorflow.core.framework.types_pb2.DataType.ValueType
+    @property
+    def shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        shape: collections.abc.Iterable[builtins.int] | None = ...,
+        dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["dtype", b"dtype", "name", b"name", "shape", b"shape"]) -> None: ...
+
+global___GraphTransferGraphInputNodeInfo = GraphTransferGraphInputNodeInfo
+
+@typing.final
+class GraphTransferGraphOutputNodeInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    SHAPE_FIELD_NUMBER: builtins.int
+    DTYPE_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    dtype: tensorflow.core.framework.types_pb2.DataType.ValueType
+    @property
+    def shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        shape: collections.abc.Iterable[builtins.int] | None = ...,
+        dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["dtype", b"dtype", "name", b"name", "shape", b"shape"]) -> None: ...
+
+global___GraphTransferGraphOutputNodeInfo = GraphTransferGraphOutputNodeInfo
+
+@typing.final
+class GraphTransferInfo(google.protobuf.message.Message):
+    """Protocol buffer representing a handle to a tensorflow resource. Handles are
+    not valid across executions, but can be serialized back and forth from within
+    a single run.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _Destination:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _DestinationEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GraphTransferInfo._Destination.ValueType], builtins.type
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        NOP: GraphTransferInfo._Destination.ValueType  # 0
+        HEXAGON: GraphTransferInfo._Destination.ValueType  # 1
+
+    class Destination(_Destination, metaclass=_DestinationEnumTypeWrapper): ...
+    NOP: GraphTransferInfo.Destination.ValueType  # 0
+    HEXAGON: GraphTransferInfo.Destination.ValueType  # 1
+
+    NODE_INFO_FIELD_NUMBER: builtins.int
+    CONST_NODE_INFO_FIELD_NUMBER: builtins.int
+    NODE_INPUT_INFO_FIELD_NUMBER: builtins.int
+    NODE_OUTPUT_INFO_FIELD_NUMBER: builtins.int
+    GRAPH_INPUT_NODE_INFO_FIELD_NUMBER: builtins.int
+    GRAPH_OUTPUT_NODE_INFO_FIELD_NUMBER: builtins.int
+    DESTINATION_FIELD_NUMBER: builtins.int
+    destination: global___GraphTransferInfo.Destination.ValueType
+    """Destination of graph transfer"""
+    @property
+    def node_info(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInfo]: ...
+    @property
+    def const_node_info(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferConstNodeInfo]: ...
+    @property
+    def node_input_info(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInputInfo]: ...
+    @property
+    def node_output_info(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeOutputInfo]: ...
+    @property
+    def graph_input_node_info(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferGraphInputNodeInfo]:
+        """Input Node parameters of transferred graph"""
+
+    @property
+    def graph_output_node_info(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferGraphOutputNodeInfo]: ...
+    def __init__(
+        self,
+        *,
+        node_info: collections.abc.Iterable[global___GraphTransferNodeInfo] | None = ...,
+        const_node_info: collections.abc.Iterable[global___GraphTransferConstNodeInfo] | None = ...,
+        node_input_info: collections.abc.Iterable[global___GraphTransferNodeInputInfo] | None = ...,
+        node_output_info: collections.abc.Iterable[global___GraphTransferNodeOutputInfo] | None = ...,
+        graph_input_node_info: collections.abc.Iterable[global___GraphTransferGraphInputNodeInfo] | None = ...,
+        graph_output_node_info: collections.abc.Iterable[global___GraphTransferGraphOutputNodeInfo] | None = ...,
+        destination: global___GraphTransferInfo.Destination.ValueType | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "const_node_info",
+            b"const_node_info",
+            "destination",
+            b"destination",
+            "graph_input_node_info",
+            b"graph_input_node_info",
+            "graph_output_node_info",
+            b"graph_output_node_info",
+            "node_info",
+            b"node_info",
+            "node_input_info",
+            b"node_input_info",
+            "node_output_info",
+            b"node_output_info",
+        ],
+    ) -> None: ...
+
+global___GraphTransferInfo = GraphTransferInfo

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/framework/kernel_def_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.attr_value_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class KernelDef(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class AttrConstraint(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        NAME_FIELD_NUMBER: builtins.int
+        ALLOWED_VALUES_FIELD_NUMBER: builtins.int
+        name: builtins.str
+        """Name of an attr from the Op."""
+        @property
+        def allowed_values(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue:
+            """A list of values that this kernel supports for this attr.
+            Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
+            """
+
+        def __init__(
+            self,
+            *,
+            name: builtins.str | None = ...,
+            allowed_values: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["allowed_values", b"allowed_values"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["allowed_values", b"allowed_values", "name", b"name"]) -> None: ...
+
+    OP_FIELD_NUMBER: builtins.int
+    DEVICE_TYPE_FIELD_NUMBER: builtins.int
+    CONSTRAINT_FIELD_NUMBER: builtins.int
+    HOST_MEMORY_ARG_FIELD_NUMBER: builtins.int
+    LABEL_FIELD_NUMBER: builtins.int
+    PRIORITY_FIELD_NUMBER: builtins.int
+    op: builtins.str
+    """Must match the name of an Op."""
+    device_type: builtins.str
+    """Type of device this kernel runs on."""
+    label: builtins.str
+    """This allows experimental kernels to be registered for an op that
+    won't be used unless the user specifies a "_kernel" attr with
+    value matching this.
+    """
+    priority: builtins.int
+    """Prioritization of kernel amongst different devices. By default we assume
+    priority is 0. The higher the priority the better. By default (i.e. if
+    this is not set), we prefer GPU kernels over CPU.
+    """
+    @property
+    def constraint(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KernelDef.AttrConstraint]: ...
+    @property
+    def host_memory_arg(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Names of the Op's input_/output_args that reside in host memory
+        instead of device memory.
+        """
+
+    def __init__(
+        self,
+        *,
+        op: builtins.str | None = ...,
+        device_type: builtins.str | None = ...,
+        constraint: collections.abc.Iterable[global___KernelDef.AttrConstraint] | None = ...,
+        host_memory_arg: collections.abc.Iterable[builtins.str] | None = ...,
+        label: builtins.str | None = ...,
+        priority: builtins.int | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "constraint",
+            b"constraint",
+            "device_type",
+            b"device_type",
+            "host_memory_arg",
+            b"host_memory_arg",
+            "label",
+            b"label",
+            "op",
+            b"op",
+            "priority",
+            b"priority",
+        ],
+    ) -> None: ...
+
+global___KernelDef = KernelDef
+
+@typing.final
+class KernelList(google.protobuf.message.Message):
+    """A collection of KernelDefs"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    KERNEL_FIELD_NUMBER: builtins.int
+    @property
+    def kernel(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KernelDef]: ...
+    def __init__(self, *, kernel: collections.abc.Iterable[global___KernelDef] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["kernel", b"kernel"]) -> None: ...
+
+global___KernelList = KernelList

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

@@ -0,0 +1,218 @@
+"""
+@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.tensor_description_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class MemoryLogStep(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STEP_ID_FIELD_NUMBER: builtins.int
+    HANDLE_FIELD_NUMBER: builtins.int
+    step_id: builtins.int
+    """Process-unique step id."""
+    handle: builtins.str
+    """Handle describing the feeds and fetches of the step."""
+    def __init__(self, *, step_id: builtins.int | None = ..., handle: builtins.str | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["handle", b"handle", "step_id", b"step_id"]) -> None: ...
+
+global___MemoryLogStep = MemoryLogStep
+
+@typing.final
+class MemoryLogTensorAllocation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STEP_ID_FIELD_NUMBER: builtins.int
+    KERNEL_NAME_FIELD_NUMBER: builtins.int
+    TENSOR_FIELD_NUMBER: builtins.int
+    step_id: builtins.int
+    """Process-unique step id."""
+    kernel_name: builtins.str
+    """Name of the kernel making the allocation as set in GraphDef,
+    e.g., "affine2/weights/Assign".
+    """
+    @property
+    def tensor(self) -> tensorflow.core.framework.tensor_description_pb2.TensorDescription:
+        """Allocated tensor details."""
+
+    def __init__(
+        self,
+        *,
+        step_id: builtins.int | None = ...,
+        kernel_name: builtins.str | None = ...,
+        tensor: tensorflow.core.framework.tensor_description_pb2.TensorDescription | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["tensor", b"tensor"]) -> builtins.bool: ...
+    def ClearField(
+        self, field_name: typing.Literal["kernel_name", b"kernel_name", "step_id", b"step_id", "tensor", b"tensor"]
+    ) -> None: ...
+
+global___MemoryLogTensorAllocation = MemoryLogTensorAllocation
+
+@typing.final
+class MemoryLogTensorDeallocation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ALLOCATION_ID_FIELD_NUMBER: builtins.int
+    ALLOCATOR_NAME_FIELD_NUMBER: builtins.int
+    allocation_id: builtins.int
+    """Id of the tensor buffer being deallocated, used to match to a
+    corresponding allocation.
+    """
+    allocator_name: builtins.str
+    """Name of the allocator used."""
+    def __init__(self, *, allocation_id: builtins.int | None = ..., allocator_name: builtins.str | None = ...) -> None: ...
+    def ClearField(
+        self, field_name: typing.Literal["allocation_id", b"allocation_id", "allocator_name", b"allocator_name"]
+    ) -> None: ...
+
+global___MemoryLogTensorDeallocation = MemoryLogTensorDeallocation
+
+@typing.final
+class MemoryLogTensorOutput(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STEP_ID_FIELD_NUMBER: builtins.int
+    KERNEL_NAME_FIELD_NUMBER: builtins.int
+    INDEX_FIELD_NUMBER: builtins.int
+    TENSOR_FIELD_NUMBER: builtins.int
+    step_id: builtins.int
+    """Process-unique step id."""
+    kernel_name: builtins.str
+    """Name of the kernel producing an output as set in GraphDef, e.g.,
+    "affine2/weights/Assign".
+    """
+    index: builtins.int
+    """Index of the output being set."""
+    @property
+    def tensor(self) -> tensorflow.core.framework.tensor_description_pb2.TensorDescription:
+        """Output tensor details."""
+
+    def __init__(
+        self,
+        *,
+        step_id: builtins.int | None = ...,
+        kernel_name: builtins.str | None = ...,
+        index: builtins.int | None = ...,
+        tensor: tensorflow.core.framework.tensor_description_pb2.TensorDescription | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["tensor", b"tensor"]) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal["index", b"index", "kernel_name", b"kernel_name", "step_id", b"step_id", "tensor", b"tensor"],
+    ) -> None: ...
+
+global___MemoryLogTensorOutput = MemoryLogTensorOutput
+
+@typing.final
+class MemoryLogRawAllocation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STEP_ID_FIELD_NUMBER: builtins.int
+    OPERATION_FIELD_NUMBER: builtins.int
+    NUM_BYTES_FIELD_NUMBER: builtins.int
+    PTR_FIELD_NUMBER: builtins.int
+    ALLOCATION_ID_FIELD_NUMBER: builtins.int
+    ALLOCATOR_NAME_FIELD_NUMBER: builtins.int
+    step_id: builtins.int
+    """Process-unique step id."""
+    operation: builtins.str
+    """Name of the operation making the allocation."""
+    num_bytes: builtins.int
+    """Number of bytes in the allocation."""
+    ptr: builtins.int
+    """Address of the allocation."""
+    allocation_id: builtins.int
+    """Id of the tensor buffer being allocated, used to match to a
+    corresponding deallocation.
+    """
+    allocator_name: builtins.str
+    """Name of the allocator used."""
+    def __init__(
+        self,
+        *,
+        step_id: builtins.int | None = ...,
+        operation: builtins.str | None = ...,
+        num_bytes: builtins.int | None = ...,
+        ptr: builtins.int | None = ...,
+        allocation_id: builtins.int | None = ...,
+        allocator_name: builtins.str | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "allocation_id",
+            b"allocation_id",
+            "allocator_name",
+            b"allocator_name",
+            "num_bytes",
+            b"num_bytes",
+            "operation",
+            b"operation",
+            "ptr",
+            b"ptr",
+            "step_id",
+            b"step_id",
+        ],
+    ) -> None: ...
+
+global___MemoryLogRawAllocation = MemoryLogRawAllocation
+
+@typing.final
+class MemoryLogRawDeallocation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STEP_ID_FIELD_NUMBER: builtins.int
+    OPERATION_FIELD_NUMBER: builtins.int
+    ALLOCATION_ID_FIELD_NUMBER: builtins.int
+    ALLOCATOR_NAME_FIELD_NUMBER: builtins.int
+    DEFERRED_FIELD_NUMBER: builtins.int
+    step_id: builtins.int
+    """Process-unique step id."""
+    operation: builtins.str
+    """Name of the operation making the deallocation."""
+    allocation_id: builtins.int
+    """Id of the tensor buffer being deallocated, used to match to a
+    corresponding allocation.
+    """
+    allocator_name: builtins.str
+    """Name of the allocator used."""
+    deferred: builtins.bool
+    """True if the deallocation is queued and will be performed later,
+    e.g. for GPU lazy freeing of buffers.
+    """
+    def __init__(
+        self,
+        *,
+        step_id: builtins.int | None = ...,
+        operation: builtins.str | None = ...,
+        allocation_id: builtins.int | None = ...,
+        allocator_name: builtins.str | None = ...,
+        deferred: builtins.bool | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "allocation_id",
+            b"allocation_id",
+            "allocator_name",
+            b"allocator_name",
+            "deferred",
+            b"deferred",
+            "operation",
+            b"operation",
+            "step_id",
+            b"step_id",
+        ],
+    ) -> None: ...
+
+global___MemoryLogRawDeallocation = MemoryLogRawDeallocation