basedpyright 1.36.2-e608e134fa492017025769bb76ecda091269eba2 → 1.37.0-532288fbedb4987e3338cbe344211c2af739a95a

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_profile_printer_data_pb2.pyi

@@ -0,0 +1,187 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2018 The OpenXLA Authors.
+
+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 HloProfilePrinterData(google.protobuf.message.Message):
+    """Describes how to pretty-print a profile counter array gathered for a specific
+    HloModule.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class HloInstructionInfo(google.protobuf.message.Message):
+        """Pretty-printer information about an HloInstruction."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        LONG_NAME_FIELD_NUMBER: builtins.int
+        SHORT_NAME_FIELD_NUMBER: builtins.int
+        CATEGORY_FIELD_NUMBER: builtins.int
+        FLOP_COUNT_FIELD_NUMBER: builtins.int
+        TRANSCENDENTAL_COUNT_FIELD_NUMBER: builtins.int
+        BYTES_ACCESSED_FIELD_NUMBER: builtins.int
+        OPTIMAL_SECONDS_FIELD_NUMBER: builtins.int
+        PROFILE_INDEX_FIELD_NUMBER: builtins.int
+        long_name: builtins.str
+        short_name: builtins.str
+        category: builtins.str
+        flop_count: builtins.float
+        """Metrics computed by HloCostAnalysis."""
+        transcendental_count: builtins.float
+        bytes_accessed: builtins.int
+        optimal_seconds: builtins.float
+        profile_index: builtins.int
+        """The index into the profile counters array for the HloInstruction
+        corresponding to this HloInstructionInfo.
+        """
+        def __init__(
+            self,
+            *,
+            long_name: builtins.str | None = ...,
+            short_name: builtins.str | None = ...,
+            category: builtins.str | None = ...,
+            flop_count: builtins.float | None = ...,
+            transcendental_count: builtins.float | None = ...,
+            bytes_accessed: builtins.int | None = ...,
+            optimal_seconds: builtins.float | None = ...,
+            profile_index: builtins.int | None = ...,
+        ) -> None: ...
+        def ClearField(
+            self,
+            field_name: typing.Literal[
+                "bytes_accessed",
+                b"bytes_accessed",
+                "category",
+                b"category",
+                "flop_count",
+                b"flop_count",
+                "long_name",
+                b"long_name",
+                "optimal_seconds",
+                b"optimal_seconds",
+                "profile_index",
+                b"profile_index",
+                "short_name",
+                b"short_name",
+                "transcendental_count",
+                b"transcendental_count",
+            ],
+        ) -> None: ...
+
+    @typing.final
+    class HloComputationInfo(google.protobuf.message.Message):
+        """Pretty-printer information about an HloComputation."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        NAME_FIELD_NUMBER: builtins.int
+        PROFILE_INDEX_FIELD_NUMBER: builtins.int
+        INSTRUCTION_INFOS_FIELD_NUMBER: builtins.int
+        name: builtins.str
+        profile_index: builtins.int
+        """The index into the profile counters array for the HloComputation
+        corresponding to this HloComputationInfo.
+        """
+        @property
+        def instruction_infos(
+            self,
+        ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
+            global___HloProfilePrinterData.HloInstructionInfo
+        ]:
+            """HloInstructionInfos for every HloInstruction in the HloComputation for
+            corresponding to this HloComputattionInfo.
+            """
+
+        def __init__(
+            self,
+            *,
+            name: builtins.str | None = ...,
+            profile_index: builtins.int | None = ...,
+            instruction_infos: collections.abc.Iterable[global___HloProfilePrinterData.HloInstructionInfo] | None = ...,
+        ) -> None: ...
+        def ClearField(
+            self,
+            field_name: typing.Literal[
+                "instruction_infos", b"instruction_infos", "name", b"name", "profile_index", b"profile_index"
+            ],
+        ) -> None: ...
+
+    @typing.final
+    class ExtraMetricsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.int
+        def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    COMPUTATION_INFOS_FIELD_NUMBER: builtins.int
+    PROFILE_COUNTERS_SIZE_FIELD_NUMBER: builtins.int
+    EXTRA_METRICS_FIELD_NUMBER: builtins.int
+    ENTRY_COMPUTATION_FIELD_NUMBER: builtins.int
+    profile_counters_size: builtins.int
+    """The size of the profile counters array we will pretty-print."""
+    entry_computation: builtins.str
+    """Name of the entry computation."""
+    @property
+    def computation_infos(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloProfilePrinterData.HloComputationInfo]:
+        """HloComputationInfos for every HloComputation in the HloModule."""
+
+    @property
+    def extra_metrics(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.int]:
+        """Maps extra metric name to the index into the profile counters array."""
+
+    def __init__(
+        self,
+        *,
+        computation_infos: collections.abc.Iterable[global___HloProfilePrinterData.HloComputationInfo] | None = ...,
+        profile_counters_size: builtins.int | None = ...,
+        extra_metrics: collections.abc.Mapping[builtins.str, builtins.int] | None = ...,
+        entry_computation: builtins.str | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "computation_infos",
+            b"computation_infos",
+            "entry_computation",
+            b"entry_computation",
+            "extra_metrics",
+            b"extra_metrics",
+            "profile_counters_size",
+            b"profile_counters_size",
+        ],
+    ) -> None: ...
+
+global___HloProfilePrinterData = HloProfilePrinterData

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/compiler/xla/service/metrics_pb2.pyi

@@ -0,0 +1,283 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import sys
+import typing
+
+import google.protobuf.any_pb2
+import google.protobuf.descriptor
+import google.protobuf.duration_pb2
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class PassMetrics(google.protobuf.message.Message):
+    """Defines pass specific metrics."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODULE_ID_FIELD_NUMBER: builtins.int
+    PASS_NAME_FIELD_NUMBER: builtins.int
+    PASS_DURATION_FIELD_NUMBER: builtins.int
+    CUSTOM_METRICS_FIELD_NUMBER: builtins.int
+    module_id: builtins.int
+    """Unique ID of the module on which the pass was run."""
+    pass_name: builtins.str
+    """The name of the pass."""
+    @property
+    def pass_duration(self) -> google.protobuf.duration_pb2.Duration:
+        """Duration of the pass."""
+
+    @property
+    def custom_metrics(self) -> google.protobuf.any_pb2.Any:
+        """Custom pass metrics. This is kept opaque, via `google.protobuf.Any`, in
+        order to decouple pass agnostic compilation logs from possibly proprietary
+        compiler passes.
+        """
+
+    def __init__(
+        self,
+        *,
+        module_id: builtins.int | None = ...,
+        pass_name: builtins.str | None = ...,
+        pass_duration: google.protobuf.duration_pb2.Duration | None = ...,
+        custom_metrics: google.protobuf.any_pb2.Any | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["custom_metrics", b"custom_metrics", "pass_duration", b"pass_duration"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "custom_metrics",
+            b"custom_metrics",
+            "module_id",
+            b"module_id",
+            "pass_duration",
+            b"pass_duration",
+            "pass_name",
+            b"pass_name",
+        ],
+    ) -> None: ...
+
+global___PassMetrics = PassMetrics
+
+@typing.final
+class JobInfo(google.protobuf.message.Message):
+    """Defines compilation job information."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    CELL_FIELD_NUMBER: builtins.int
+    USER_FIELD_NUMBER: builtins.int
+    UID_FIELD_NUMBER: builtins.int
+    TASK_ID_FIELD_NUMBER: builtins.int
+    TASK_UID_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Name of the job running compilation."""
+    cell: builtins.str
+    """Cell in which the job is running."""
+    user: builtins.str
+    """User running the job."""
+    uid: builtins.int
+    """Unique id when combined with user and cell field."""
+    task_id: builtins.int
+    """Task index, which will not change across job restarts."""
+    task_uid: builtins.int
+    """Task unique id, which may change across job restarts."""
+    def __init__(
+        self,
+        *,
+        name: builtins.str | None = ...,
+        cell: builtins.str | None = ...,
+        user: builtins.str | None = ...,
+        uid: builtins.int | None = ...,
+        task_id: builtins.int | None = ...,
+        task_uid: builtins.int | None = ...,
+    ) -> None: ...
+    def HasField(
+        self,
+        field_name: typing.Literal[
+            "_cell",
+            b"_cell",
+            "_name",
+            b"_name",
+            "_task_id",
+            b"_task_id",
+            "_task_uid",
+            b"_task_uid",
+            "_uid",
+            b"_uid",
+            "_user",
+            b"_user",
+            "cell",
+            b"cell",
+            "name",
+            b"name",
+            "task_id",
+            b"task_id",
+            "task_uid",
+            b"task_uid",
+            "uid",
+            b"uid",
+            "user",
+            b"user",
+        ],
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "_cell",
+            b"_cell",
+            "_name",
+            b"_name",
+            "_task_id",
+            b"_task_id",
+            "_task_uid",
+            b"_task_uid",
+            "_uid",
+            b"_uid",
+            "_user",
+            b"_user",
+            "cell",
+            b"cell",
+            "name",
+            b"name",
+            "task_id",
+            b"task_id",
+            "task_uid",
+            b"task_uid",
+            "uid",
+            b"uid",
+            "user",
+            b"user",
+        ],
+    ) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_cell", b"_cell"]) -> typing.Literal["cell"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_name", b"_name"]) -> typing.Literal["name"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_task_id", b"_task_id"]) -> typing.Literal["task_id"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_task_uid", b"_task_uid"]) -> typing.Literal["task_uid"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_uid", b"_uid"]) -> typing.Literal["uid"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_user", b"_user"]) -> typing.Literal["user"] | None: ...
+
+global___JobInfo = JobInfo
+
+@typing.final
+class CompilationLogEntry(google.protobuf.message.Message):
+    """Defines XLA compilation metrics."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _CompilationStage:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _CompilationStageEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CompilationLogEntry._CompilationStage.ValueType],
+        builtins.type,
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        UNSPECIFIED: CompilationLogEntry._CompilationStage.ValueType  # 0
+        END_TO_END: CompilationLogEntry._CompilationStage.ValueType  # 1
+        HLO_PASSES: CompilationLogEntry._CompilationStage.ValueType  # 2
+        CODE_GENERATION: CompilationLogEntry._CompilationStage.ValueType  # 3
+        BACKEND_PASSES: CompilationLogEntry._CompilationStage.ValueType  # 4
+
+    class CompilationStage(_CompilationStage, metaclass=_CompilationStageEnumTypeWrapper):
+        """Defines compilation stages for which metrics are collected."""
+
+    UNSPECIFIED: CompilationLogEntry.CompilationStage.ValueType  # 0
+    END_TO_END: CompilationLogEntry.CompilationStage.ValueType  # 1
+    HLO_PASSES: CompilationLogEntry.CompilationStage.ValueType  # 2
+    CODE_GENERATION: CompilationLogEntry.CompilationStage.ValueType  # 3
+    BACKEND_PASSES: CompilationLogEntry.CompilationStage.ValueType  # 4
+
+    TIMESTAMP_FIELD_NUMBER: builtins.int
+    STAGE_FIELD_NUMBER: builtins.int
+    DURATION_FIELD_NUMBER: builtins.int
+    TASK_INDEX_FIELD_NUMBER: builtins.int
+    PASS_METRICS_FIELD_NUMBER: builtins.int
+    MODULE_IDS_FIELD_NUMBER: builtins.int
+    JOB_INFO_FIELD_NUMBER: builtins.int
+    stage: global___CompilationLogEntry.CompilationStage.ValueType
+    """Compilation stage recorded by this log entry."""
+    task_index: builtins.int
+    """Task index from which this log entry was recorded or
+    -1 if the task index could not be fetched. In the case task_index is not
+    equal to -1, it is guaranteed to match the task_id in job_info.
+    """
+    @property
+    def timestamp(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Time when the event captured by this log entry occurred."""
+
+    @property
+    def duration(self) -> google.protobuf.duration_pb2.Duration:
+        """Duration of the given compilation stage."""
+
+    @property
+    def pass_metrics(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PassMetrics]:
+        """Pass specific metrics."""
+
+    @property
+    def module_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """IDs of modules on which the compilation stage was run."""
+
+    @property
+    def job_info(self) -> global___JobInfo:
+        """Job information."""
+
+    def __init__(
+        self,
+        *,
+        timestamp: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        stage: global___CompilationLogEntry.CompilationStage.ValueType | None = ...,
+        duration: google.protobuf.duration_pb2.Duration | None = ...,
+        task_index: builtins.int | None = ...,
+        pass_metrics: collections.abc.Iterable[global___PassMetrics] | None = ...,
+        module_ids: collections.abc.Iterable[builtins.int] | None = ...,
+        job_info: global___JobInfo | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["duration", b"duration", "job_info", b"job_info", "timestamp", b"timestamp"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "duration",
+            b"duration",
+            "job_info",
+            b"job_info",
+            "module_ids",
+            b"module_ids",
+            "pass_metrics",
+            b"pass_metrics",
+            "stage",
+            b"stage",
+            "task_index",
+            b"task_index",
+            "timestamp",
+            b"timestamp",
+        ],
+    ) -> None: ...
+
+global___CompilationLogEntry = CompilationLogEntry

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/compiler/xla/service/test_compilation_environment_pb2.pyi

@@ -0,0 +1,59 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2022 The OpenXLA Authors.
+
+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 typing
+
+import google.protobuf.descriptor
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class TestCompilationEnvironment1(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOME_FLAG_FIELD_NUMBER: builtins.int
+    some_flag: builtins.int
+    def __init__(self, *, some_flag: builtins.int | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["some_flag", b"some_flag"]) -> None: ...
+
+global___TestCompilationEnvironment1 = TestCompilationEnvironment1
+
+@typing.final
+class TestCompilationEnvironment2(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOME_OTHER_FLAG_FIELD_NUMBER: builtins.int
+    some_other_flag: builtins.int
+    def __init__(self, *, some_other_flag: builtins.int | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["some_other_flag", b"some_other_flag"]) -> None: ...
+
+global___TestCompilationEnvironment2 = TestCompilationEnvironment2
+
+@typing.final
+class TestCompilationEnvironment3(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    A_THIRD_FLAG_FIELD_NUMBER: builtins.int
+    a_third_flag: builtins.int
+    def __init__(self, *, a_third_flag: builtins.int | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["a_third_flag", b"a_third_flag"]) -> None: ...
+
+global___TestCompilationEnvironment3 = TestCompilationEnvironment3

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/compiler/xla/service/xla_compile_result_pb2.pyi

@@ -0,0 +1,167 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2023 The OpenXLA Authors.
+
+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.duration_pb2
+import google.protobuf.internal.containers
+import google.protobuf.message
+import tensorflow.compiler.xla.service.hlo_pb2
+import tensorflow.tsl.protobuf.status_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class CompilerPerfStats(google.protobuf.message.Message):
+    """Statistics on how long various parts of compilation took.
+    Not all durations may be relevant for all producers of this message, in
+    which irrelevant fields should simply be skipped.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    INIT_DURATION_FIELD_NUMBER: builtins.int
+    HLO_VERIFICATION_DURATION_FIELD_NUMBER: builtins.int
+    COMPILATION_PROLOGUE_DURATION_FIELD_NUMBER: builtins.int
+    COMPILATION_DURATION_FIELD_NUMBER: builtins.int
+    TOTAL_DURATION_FIELD_NUMBER: builtins.int
+    @property
+    def init_duration(self) -> google.protobuf.duration_pb2.Duration:
+        """How long did it take to initialize the compiler?"""
+
+    @property
+    def hlo_verification_duration(self) -> google.protobuf.duration_pb2.Duration:
+        """How long did it take to verify the HLO?"""
+
+    @property
+    def compilation_prologue_duration(self) -> google.protobuf.duration_pb2.Duration:
+        """How long did it take to prepare for compilation after verification?"""
+
+    @property
+    def compilation_duration(self) -> google.protobuf.duration_pb2.Duration:
+        """How long did it take to compile?"""
+
+    @property
+    def total_duration(self) -> google.protobuf.duration_pb2.Duration:
+        """How long did everything take?"""
+
+    def __init__(
+        self,
+        *,
+        init_duration: google.protobuf.duration_pb2.Duration | None = ...,
+        hlo_verification_duration: google.protobuf.duration_pb2.Duration | None = ...,
+        compilation_prologue_duration: google.protobuf.duration_pb2.Duration | None = ...,
+        compilation_duration: google.protobuf.duration_pb2.Duration | None = ...,
+        total_duration: google.protobuf.duration_pb2.Duration | None = ...,
+    ) -> None: ...
+    def HasField(
+        self,
+        field_name: typing.Literal[
+            "compilation_duration",
+            b"compilation_duration",
+            "compilation_prologue_duration",
+            b"compilation_prologue_duration",
+            "hlo_verification_duration",
+            b"hlo_verification_duration",
+            "init_duration",
+            b"init_duration",
+            "total_duration",
+            b"total_duration",
+        ],
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "compilation_duration",
+            b"compilation_duration",
+            "compilation_prologue_duration",
+            b"compilation_prologue_duration",
+            "hlo_verification_duration",
+            b"hlo_verification_duration",
+            "init_duration",
+            b"init_duration",
+            "total_duration",
+            b"total_duration",
+        ],
+    ) -> None: ...
+
+global___CompilerPerfStats = CompilerPerfStats
+
+@typing.final
+class CompilationResult(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class CountersEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.int
+        def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ...
+        def HasField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    HLO_MODULE_FIELD_NUMBER: builtins.int
+    PERF_STATS_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    COUNTERS_FIELD_NUMBER: builtins.int
+    @property
+    def hlo_module(self) -> tensorflow.compiler.xla.service.hlo_pb2.HloModuleProto:
+        """The compiled HLO. Only set when compilation succeeds."""
+
+    @property
+    def perf_stats(self) -> global___CompilerPerfStats:
+        """Always set when compilation succeeds. May or may not be set when
+        compilation fails.
+        """
+
+    @property
+    def status(self) -> tensorflow.tsl.protobuf.status_pb2.StatusProto:
+        """Always set even when compilation succeeds."""
+
+    @property
+    def counters(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.int]:
+        """Collects counters collected during compilation. Not every producer may
+        include counter support at all or any particular counter.
+        """
+
+    def __init__(
+        self,
+        *,
+        hlo_module: tensorflow.compiler.xla.service.hlo_pb2.HloModuleProto | None = ...,
+        perf_stats: global___CompilerPerfStats | None = ...,
+        status: tensorflow.tsl.protobuf.status_pb2.StatusProto | None = ...,
+        counters: collections.abc.Mapping[builtins.str, builtins.int] | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["hlo_module", b"hlo_module", "perf_stats", b"perf_stats", "status", b"status"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "counters", b"counters", "hlo_module", b"hlo_module", "perf_stats", b"perf_stats", "status", b"status"
+        ],
+    ) -> None: ...
+
+global___CompilationResult = CompilationResult