caffe2/proto/prof_dag_pb2.pyi

"""
@generated by mypy-protobuf.  Do not edit manually!
isort:skip_file
"""
import builtins
import google.protobuf.descriptor
import google.protobuf.internal.containers
import google.protobuf.message
import typing
import typing_extensions

DESCRIPTOR: google.protobuf.descriptor.FileDescriptor = ...

class TwoNumberStatsProto(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor = ...
    MEAN_FIELD_NUMBER: int
    STDDEV_FIELD_NUMBER: int
    COUNT_FIELD_NUMBER: int
    mean: float = ...
    stddev: float = ...
    count: int = ...

    def __init__(self,
        *,
        mean : typing.Optional[float] = ...,
        stddev : typing.Optional[float] = ...,
        count : typing.Optional[int] = ...,
        ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal[u"count",b"count",u"mean",b"mean",u"stddev",b"stddev"]) -> bool: ...
    def ClearField(self, field_name: typing_extensions.Literal[u"count",b"count",u"mean",b"mean",u"stddev",b"stddev"]) -> None: ...
global___TwoNumberStatsProto = TwoNumberStatsProto

class BlobProfile(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor = ...
    NAME_FIELD_NUMBER: int
    BYTES_USED_FIELD_NUMBER: int
    name: typing.Text = ...

    @property
    def bytes_used(self) -> global___TwoNumberStatsProto: ...

    def __init__(self,
        *,
        name : typing.Optional[typing.Text] = ...,
        bytes_used : typing.Optional[global___TwoNumberStatsProto] = ...,
        ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal[u"bytes_used",b"bytes_used",u"name",b"name"]) -> bool: ...
    def ClearField(self, field_name: typing_extensions.Literal[u"bytes_used",b"bytes_used",u"name",b"name"]) -> None: ...
global___BlobProfile = BlobProfile

class ProfDAGProto(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor = ...
    NAME_FIELD_NUMBER: int
    MEAN_FIELD_NUMBER: int
    STDDEV_FIELD_NUMBER: int
    EXECUTION_TIME_FIELD_NUMBER: int
    OUTPUT_PROFILE_FIELD_NUMBER: int
    EXTRA_INFO_FIELD_NUMBER: int
    name: typing.Text = ...
    mean: float = ...
    stddev: float = ...
    extra_info: google.protobuf.internal.containers.RepeatedScalarFieldContainer[typing.Text] = ...

    @property
    def execution_time(self) -> global___TwoNumberStatsProto: ...

    @property
    def output_profile(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BlobProfile]: ...

    def __init__(self,
        *,
        name : typing.Optional[typing.Text] = ...,
        mean : typing.Optional[float] = ...,
        stddev : typing.Optional[float] = ...,
        execution_time : typing.Optional[global___TwoNumberStatsProto] = ...,
        output_profile : typing.Optional[typing.Iterable[global___BlobProfile]] = ...,
        extra_info : typing.Optional[typing.Iterable[typing.Text]] = ...,
        ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal[u"execution_time",b"execution_time",u"mean",b"mean",u"name",b"name",u"stddev",b"stddev"]) -> bool: ...
    def ClearField(self, field_name: typing_extensions.Literal[u"execution_time",b"execution_time",u"extra_info",b"extra_info",u"mean",b"mean",u"name",b"name",u"output_profile",b"output_profile",u"stddev",b"stddev"]) -> None: ...
global___ProfDAGProto = ProfDAGProto

class ProfDAGProtos(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor = ...
    STATS_FIELD_NUMBER: int
    NET_NAME_FIELD_NUMBER: int
    OPS_STATS_FIELD_NUMBER: int
    net_name: typing.Text = ...

    @property
    def stats(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ProfDAGProto]: ...

    @property
    def ops_stats(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpProfile]: ...

    def __init__(self,
        *,
        stats : typing.Optional[typing.Iterable[global___ProfDAGProto]] = ...,
        net_name : typing.Optional[typing.Text] = ...,
        ops_stats : typing.Optional[typing.Iterable[global___OpProfile]] = ...,
        ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal[u"net_name",b"net_name"]) -> bool: ...
    def ClearField(self, field_name: typing_extensions.Literal[u"net_name",b"net_name",u"ops_stats",b"ops_stats",u"stats",b"stats"]) -> None: ...
global___ProfDAGProtos = ProfDAGProtos

class OpProfile(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor = ...
    IDX_FIELD_NUMBER: int
    NET_NAME_FIELD_NUMBER: int
    TYPE_FIELD_NUMBER: int
    EXEC_TIME_SECS_FIELD_NUMBER: int
    idx: typing.Text = ...
    net_name: typing.Text = ...
    type: typing.Text = ...
    exec_time_secs: float = ...

    def __init__(self,
        *,
        idx : typing.Optional[typing.Text] = ...,
        net_name : typing.Optional[typing.Text] = ...,
        type : typing.Optional[typing.Text] = ...,
        exec_time_secs : typing.Optional[float] = ...,
        ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal[u"exec_time_secs",b"exec_time_secs",u"idx",b"idx",u"net_name",b"net_name",u"type",b"type"]) -> bool: ...
    def ClearField(self, field_name: typing_extensions.Literal[u"exec_time_secs",b"exec_time_secs",u"idx",b"idx",u"net_name",b"net_name",u"type",b"type"]) -> None: ...
global___OpProfile = OpProfile