xprof-nightly 2.22.3a20251208__cp311-none-manylinux2014_x86_64.whl

xprof/protobuf/source_stats_pb2.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/source_stats.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/source_stats.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(plugin/xprof/protobuf/source_stats.proto\x12\x13tensorflow.profiler\"\x95\x04\n\x0bSourceStats\x12U\n\x14\x66ile_name_to_metrics\x18\x01 \x03(\x0b\x32\x37.tensorflow.profiler.SourceStats.FileNameToMetricsEntry\x1an\n\x06Metric\x12\x13\n\x0boccurrences\x18\x01 \x01(\x04\x12\x14\n\x0cself_time_ps\x18\x02 \x01(\x04\x12\x0f\n\x07time_ps\x18\x03 \x01(\x04\x12\r\n\x05\x66lops\x18\x04 \x01(\x04\x12\x19\n\x11\x66lops_utilization\x18\x05 \x01(\x01\x1a\xd6\x01\n\x0b\x46ileMetrics\x12\x63\n\x15line_number_to_metric\x18\x01 \x03(\x0b\x32\x44.tensorflow.profiler.SourceStats.FileMetrics.LineNumberToMetricEntry\x1a\x62\n\x17LineNumberToMetricEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.tensorflow.profiler.SourceStats.Metric:\x02\x38\x01\x1a\x66\n\x16\x46ileNameToMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12;\n\x05value\x18\x02 \x01(\x0b\x32,.tensorflow.profiler.SourceStats.FileMetrics:\x02\x38\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.source_stats_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_SOURCESTATS_FILEMETRICS_LINENUMBERTOMETRICENTRY']._loaded_options = None
+  _globals['_SOURCESTATS_FILEMETRICS_LINENUMBERTOMETRICENTRY']._serialized_options = b'8\001'
+  _globals['_SOURCESTATS_FILENAMETOMETRICSENTRY']._loaded_options = None
+  _globals['_SOURCESTATS_FILENAMETOMETRICSENTRY']._serialized_options = b'8\001'
+  _globals['_SOURCESTATS']._serialized_start=66
+  _globals['_SOURCESTATS']._serialized_end=599
+  _globals['_SOURCESTATS_METRIC']._serialized_start=168
+  _globals['_SOURCESTATS_METRIC']._serialized_end=278
+  _globals['_SOURCESTATS_FILEMETRICS']._serialized_start=281
+  _globals['_SOURCESTATS_FILEMETRICS']._serialized_end=495
+  _globals['_SOURCESTATS_FILEMETRICS_LINENUMBERTOMETRICENTRY']._serialized_start=397
+  _globals['_SOURCESTATS_FILEMETRICS_LINENUMBERTOMETRICENTRY']._serialized_end=495
+  _globals['_SOURCESTATS_FILENAMETOMETRICSENTRY']._serialized_start=497
+  _globals['_SOURCESTATS_FILENAMETOMETRICSENTRY']._serialized_end=599
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/steps_db_pb2.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/steps_db.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/steps_db.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+from xprof.protobuf import op_metrics_pb2 as plugin_dot_xprof_dot_protobuf_dot_op__metrics__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$plugin/xprof/protobuf/steps_db.proto\x12\x13tensorflow.profiler\x1a\x19google/protobuf/any.proto\x1a&plugin/xprof/protobuf/op_metrics.proto\"\x90\x02\n\x14GenericStepBreakdown\x12\x46\n\x07type_ps\x18\x01 \x03(\x0b\x32\x35.tensorflow.profiler.GenericStepBreakdown.TypePsEntry\x12N\n\x0b\x63\x61tegory_ps\x18\x02 \x03(\x0b\x32\x39.tensorflow.profiler.GenericStepBreakdown.CategoryPsEntry\x1a-\n\x0bTypePsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x04:\x02\x38\x01\x1a\x31\n\x0f\x43\x61tegoryPsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x04:\x02\x38\x01\"\x9b\x04\n\x10TpuStepBreakdown\x12\x1a\n\x12infeed_duration_ps\x18\x01 \x01(\x04\x12\x17\n\x0fhost_outfeed_ps\x18\x02 \x01(\x04\x12!\n\x19wait_for_scv0_duration_ps\x18\x03 \x01(\x04\x12 \n\x18scv0_infeed_transform_ps\x18\x04 \x01(\x04\x12\x17\n\x0fscv0_outfeed_ps\x18\x05 \x01(\x04\x12\x17\n\x0f\x63rs_duration_ps\x18\x06 \x01(\x04\x12\x1b\n\x13scv0_infeed_percent\x18\x07 \x01(\x01\x12\x18\n\x10send_duration_ps\x18\x08 \x01(\x04\x12\x18\n\x10recv_duration_ps\x18\t \x01(\x04\x12\x1d\n\x15host_send_duration_ps\x18\x0f \x01(\x04\x12\x1d\n\x15host_recv_duration_ps\x18\x10 \x01(\x04\x12\x31\n)wait_for_megacore_fusion_peer_duration_ps\x18\x0e \x01(\x04\x12 \n\x18overlay_wait_duration_ps\x18\x0b \x01(\x04\x12\x1d\n\x15high_flops_compute_ps\x18\x0c \x01(\x04\x12\x12\n\ntc_idle_ps\x18\r \x01(\x04\x12\x12\n\ntc_busy_ps\x18\x11 \x01(\x04\x12\x14\n\x0cscv0_busy_ps\x18\x12 \x01(\x04\x12\x14\n\x0cscv0_step_ps\x18\x13 \x01(\x04J\x04\x08\n\x10\x0b\"\x85\x01\n\x17SparseCoreStepBreakdown\x12\x15\n\rsc_compute_ps\x18\x01 \x01(\x04\x12\x14\n\x0csc_infeed_ps\x18\x02 \x01(\x04\x12\x15\n\rsc_outfeed_ps\x18\x03 \x01(\x04\x12\x12\n\nsc_idle_ps\x18\x04 \x01(\x04\x12\x12\n\nsc_busy_ps\x18\x05 \x01(\x04\"V\n\x14\x44\x65viceMemoryTransfer\x12\x12\n\noccurrence\x18\x01 \x01(\x04\x12\x0f\n\x07time_us\x18\x02 \x01(\x01\x12\x19\n\x11\x62ytes_transferred\x18\x03 \x01(\x04\"\xca\x01\n\x0eStepInfoResult\x12\x10\n\x08step_num\x18\x01 \x01(\r\x12\x11\n\tstep_name\x18\x05 \x01(\t\x12\x13\n\x0b\x64uration_ps\x18\x02 \x01(\x04\x12\x10\n\x08\x62\x65gin_ps\x18\x03 \x01(\x04\x12,\n\x0estep_breakdown\x18\x04 \x01(\x0b\x32\x14.google.protobuf.Any\x12>\n\x0b\x63ollectives\x18\x06 \x01(\x0b\x32).tensorflow.profiler.DeviceMemoryTransfer\"\x83\x01\n\rAllReduceInfo\x12\n\n\x02id\x18\x01 \x01(\x04\x12\x10\n\x04name\x18\x02 \x01(\tB\x02\x18\x01\x12\x15\n\rall_reduce_id\x18\x03 \x01(\x04\x12\x15\n\rstart_time_ps\x18\x04 \x01(\x04\x12\x13\n\x0b\x65nd_time_ps\x18\x05 \x01(\x04\x12\x11\n\tbyte_size\x18\x06 \x01(\x04\"P\n\x11\x41llReduceDbResult\x12;\n\x0f\x61ll_reduce_info\x18\x01 \x03(\x0b\x32\".tensorflow.profiler.AllReduceInfo\"\xc4\x05\n\x0fPerCoreStepInfo\x12\x10\n\x08step_num\x18\x01 \x01(\r\x12U\n\x12step_info_per_core\x18\x02 \x03(\x0b\x32\x39.tensorflow.profiler.PerCoreStepInfo.StepInfoPerCoreEntry\x12\x38\n\x0ehlo_metrics_db\x18\x03 \x01(\x0b\x32 .tensorflow.profiler.OpMetricsDb\x12\x61\n\x19\x63ore_id_to_replica_id_map\x18\x05 \x03(\x0b\x32>.tensorflow.profiler.PerCoreStepInfo.CoreIdToReplicaIdMapEntry\x12\\\n\x16\x61ll_reduce_db_per_core\x18\x06 \x03(\x0b\x32<.tensorflow.profiler.PerCoreStepInfo.AllReduceDbPerCoreEntry\x12J\n\x17\x64\x65vice_memory_transfers\x18\x07 \x03(\x0b\x32).tensorflow.profiler.DeviceMemoryTransfer\x1a[\n\x14StepInfoPerCoreEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32#.tensorflow.profiler.StepInfoResult:\x02\x38\x01\x1a;\n\x19\x43oreIdToReplicaIdMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\r\n\x05value\x18\x02 \x01(\r:\x02\x38\x01\x1a\x61\n\x17\x41llReduceDbPerCoreEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32&.tensorflow.profiler.AllReduceDbResult:\x02\x38\x01J\x04\x08\x04\x10\x05\"\xa2\x01\n\x12StepDatabaseResult\x12;\n\rstep_sequence\x18\x01 \x03(\x0b\x32$.tensorflow.profiler.PerCoreStepInfo\x12\x1b\n\x13use_incomplete_step\x18\x02 \x01(\x08\x12\x19\n\x11num_steps_dropped\x18\x03 \x01(\r\x12\x17\n\x0f\x65mpty_intersect\x18\x04 \x01(\x08\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.steps_db_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_GENERICSTEPBREAKDOWN_TYPEPSENTRY']._loaded_options = None
+  _globals['_GENERICSTEPBREAKDOWN_TYPEPSENTRY']._serialized_options = b'8\001'
+  _globals['_GENERICSTEPBREAKDOWN_CATEGORYPSENTRY']._loaded_options = None
+  _globals['_GENERICSTEPBREAKDOWN_CATEGORYPSENTRY']._serialized_options = b'8\001'
+  _globals['_ALLREDUCEINFO'].fields_by_name['name']._loaded_options = None
+  _globals['_ALLREDUCEINFO'].fields_by_name['name']._serialized_options = b'\030\001'
+  _globals['_PERCORESTEPINFO_STEPINFOPERCOREENTRY']._loaded_options = None
+  _globals['_PERCORESTEPINFO_STEPINFOPERCOREENTRY']._serialized_options = b'8\001'
+  _globals['_PERCORESTEPINFO_COREIDTOREPLICAIDMAPENTRY']._loaded_options = None
+  _globals['_PERCORESTEPINFO_COREIDTOREPLICAIDMAPENTRY']._serialized_options = b'8\001'
+  _globals['_PERCORESTEPINFO_ALLREDUCEDBPERCOREENTRY']._loaded_options = None
+  _globals['_PERCORESTEPINFO_ALLREDUCEDBPERCOREENTRY']._serialized_options = b'8\001'
+  _globals['_GENERICSTEPBREAKDOWN']._serialized_start=129
+  _globals['_GENERICSTEPBREAKDOWN']._serialized_end=401
+  _globals['_GENERICSTEPBREAKDOWN_TYPEPSENTRY']._serialized_start=305
+  _globals['_GENERICSTEPBREAKDOWN_TYPEPSENTRY']._serialized_end=350
+  _globals['_GENERICSTEPBREAKDOWN_CATEGORYPSENTRY']._serialized_start=352
+  _globals['_GENERICSTEPBREAKDOWN_CATEGORYPSENTRY']._serialized_end=401
+  _globals['_TPUSTEPBREAKDOWN']._serialized_start=404
+  _globals['_TPUSTEPBREAKDOWN']._serialized_end=943
+  _globals['_SPARSECORESTEPBREAKDOWN']._serialized_start=946
+  _globals['_SPARSECORESTEPBREAKDOWN']._serialized_end=1079
+  _globals['_DEVICEMEMORYTRANSFER']._serialized_start=1081
+  _globals['_DEVICEMEMORYTRANSFER']._serialized_end=1167
+  _globals['_STEPINFORESULT']._serialized_start=1170
+  _globals['_STEPINFORESULT']._serialized_end=1372
+  _globals['_ALLREDUCEINFO']._serialized_start=1375
+  _globals['_ALLREDUCEINFO']._serialized_end=1506
+  _globals['_ALLREDUCEDBRESULT']._serialized_start=1508
+  _globals['_ALLREDUCEDBRESULT']._serialized_end=1588
+  _globals['_PERCORESTEPINFO']._serialized_start=1591
+  _globals['_PERCORESTEPINFO']._serialized_end=2299
+  _globals['_PERCORESTEPINFO_STEPINFOPERCOREENTRY']._serialized_start=2042
+  _globals['_PERCORESTEPINFO_STEPINFOPERCOREENTRY']._serialized_end=2133
+  _globals['_PERCORESTEPINFO_COREIDTOREPLICAIDMAPENTRY']._serialized_start=2135
+  _globals['_PERCORESTEPINFO_COREIDTOREPLICAIDMAPENTRY']._serialized_end=2194
+  _globals['_PERCORESTEPINFO_ALLREDUCEDBPERCOREENTRY']._serialized_start=2196
+  _globals['_PERCORESTEPINFO_ALLREDUCEDBPERCOREENTRY']._serialized_end=2293
+  _globals['_STEPDATABASERESULT']._serialized_start=2302
+  _globals['_STEPDATABASERESULT']._serialized_end=2464
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/task_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/task.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/task.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n plugin/xprof/protobuf/task.proto\x12\x13tensorflow.profiler\"\xf9\x02\n\x04Task\x12\x12\n\nchangelist\x18\x01 \x01(\x03\x12\x13\n\x0b\x63lean_build\x18\x02 \x01(\x08\x12\x12\n\nbuild_time\x18\x03 \x01(\x03\x12\x14\n\x0c\x62uild_target\x18\x04 \x01(\t\x12\x14\n\x0c\x63ommand_line\x18\x05 \x01(\t\x12\x12\n\nstart_time\x18\x06 \x01(\x03\x12\x14\n\x0ctask_address\x18\x07 \x01(\t\x12\x17\n\x0fprofile_time_ns\x18\x08 \x01(\x04\x12\x1b\n\x13profile_duration_ms\x18\t \x01(\r\x12\x18\n\x10host_trace_level\x18\n \x01(\r\x12\x1b\n\x13tensor_core_freq_hz\x18\x0b \x01(\x04\x12\x1b\n\x13sparse_core_freq_hz\x18\x0c \x01(\x04\x12\x13\n\x0bgtc_freq_hz\x18\r \x01(\x04\x12\x19\n\x11peak_memory_usage\x18\x0e \x01(\x04\x12\x11\n\tcpu_limit\x18\x0f \x01(\x01\x12\x11\n\tcpu_usage\x18\x10 \x01(\x01\x42\x03\xf8\x01\x01')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.task_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\370\001\001'
+  _globals['_TASK']._serialized_start=58
+  _globals['_TASK']._serialized_end=435
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/tf_data_stats_pb2.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/tf_data_stats.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/tf_data_stats.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)plugin/xprof/protobuf/tf_data_stats.proto\x12\x13tensorflow.profiler\"\x84\x01\n\x0cIteratorStat\x12\n\n\x02id\x18\x01 \x01(\x03\x12\x15\n\rstart_time_ps\x18\x02 \x01(\x03\x12\x13\n\x0b\x64uration_ps\x18\x03 \x01(\x03\x12\x14\n\x0cself_time_ps\x18\x04 \x01(\x03\x12\x13\n\x0bis_blocking\x18\x05 \x01(\x08\x12\x11\n\tnum_calls\x18\x06 \x01(\x03\"\xd6\x01\n\x10IteratorMetadata\x12\n\n\x02id\x18\x01 \x01(\x03\x12\x11\n\tparent_id\x18\x02 \x01(\x03\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x11\n\tlong_name\x18\x06 \x01(\t\x12\x10\n\x08is_async\x18\x04 \x01(\x08\x12\x41\n\x06params\x18\x05 \x03(\x0b\x32\x31.tensorflow.profiler.IteratorMetadata.ParamsEntry\x1a-\n\x0bParamsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x87\x02\n\x11InputPipelineStat\x12\x1e\n\x16\x62ottleneck_iterator_id\x18\x02 \x01(\x03\x12&\n\x1e\x62ottleneck_iterator_latency_ps\x18\x03 \x01(\x03\x12Q\n\x0eiterator_stats\x18\x01 \x03(\x0b\x32\x39.tensorflow.profiler.InputPipelineStat.IteratorStatsEntry\x1aW\n\x12IteratorStatsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32!.tensorflow.profiler.IteratorStat:\x02\x38\x01\"\xae\x01\n\x15InputPipelineMetadata\x12\n\n\x02id\x18\x01 \x01(\x03\x12J\n\x04type\x18\x02 \x01(\x0e\x32<.tensorflow.profiler.InputPipelineMetadata.InputPipelineType\x12\x0c\n\x04name\x18\x04 \x01(\t\")\n\x11InputPipelineType\x12\x08\n\x04HOST\x10\x00\x12\n\n\x06\x44\x45VICE\x10\x01J\x04\x08\x03\x10\x04\"\xe9\x01\n\x12InputPipelineStats\x12<\n\x08metadata\x18\x01 \x01(\x0b\x32*.tensorflow.profiler.InputPipelineMetadata\x12\x16\n\x0e\x61vg_latency_ps\x18\x03 \x01(\x03\x12\x16\n\x0emin_latency_ps\x18\x04 \x01(\x03\x12\x16\n\x0emax_latency_ps\x18\x05 \x01(\x03\x12\x16\n\x0enum_slow_calls\x18\x06 \x01(\x03\x12\x35\n\x05stats\x18\x02 \x03(\x0b\x32&.tensorflow.profiler.InputPipelineStat\"\xef\x02\n\x0bTfDataStats\x12Q\n\x11iterator_metadata\x18\x02 \x03(\x0b\x32\x36.tensorflow.profiler.TfDataStats.IteratorMetadataEntry\x12M\n\x0finput_pipelines\x18\x01 \x03(\x0b\x32\x34.tensorflow.profiler.TfDataStats.InputPipelinesEntry\x1a^\n\x15IteratorMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.tensorflow.profiler.IteratorMetadata:\x02\x38\x01\x1a^\n\x13InputPipelinesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.tensorflow.profiler.InputPipelineStats:\x02\x38\x01\"\xbc\x01\n\x18TfDataBottleneckAnalysis\x12\x0c\n\x04host\x18\x01 \x01(\t\x12\x16\n\x0einput_pipeline\x18\x02 \x01(\t\x12\x16\n\x0emax_latency_ps\x18\x03 \x01(\x03\x12\x15\n\riterator_name\x18\x04 \x01(\t\x12\x1a\n\x12iterator_long_name\x18\x05 \x01(\t\x12\x1b\n\x13iterator_latency_ps\x18\x07 \x01(\x03\x12\x12\n\nsuggestion\x18\x06 \x01(\t\"\xb2\x02\n\x13\x43ombinedTfDataStats\x12\x16\n\x0eis_input_bound\x18\x03 \x01(\x08\x12\x0f\n\x07summary\x18\x04 \x01(\t\x12J\n\x13\x62ottleneck_analysis\x18\x01 \x03(\x0b\x32-.tensorflow.profiler.TfDataBottleneckAnalysis\x12P\n\rtf_data_stats\x18\x02 \x03(\x0b\x32\x39.tensorflow.profiler.CombinedTfDataStats.TfDataStatsEntry\x1aT\n\x10TfDataStatsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12/\n\x05value\x18\x02 \x01(\x0b\x32 .tensorflow.profiler.TfDataStats:\x02\x38\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.tf_data_stats_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_ITERATORMETADATA_PARAMSENTRY']._loaded_options = None
+  _globals['_ITERATORMETADATA_PARAMSENTRY']._serialized_options = b'8\001'
+  _globals['_INPUTPIPELINESTAT_ITERATORSTATSENTRY']._loaded_options = None
+  _globals['_INPUTPIPELINESTAT_ITERATORSTATSENTRY']._serialized_options = b'8\001'
+  _globals['_TFDATASTATS_ITERATORMETADATAENTRY']._loaded_options = None
+  _globals['_TFDATASTATS_ITERATORMETADATAENTRY']._serialized_options = b'8\001'
+  _globals['_TFDATASTATS_INPUTPIPELINESENTRY']._loaded_options = None
+  _globals['_TFDATASTATS_INPUTPIPELINESENTRY']._serialized_options = b'8\001'
+  _globals['_COMBINEDTFDATASTATS_TFDATASTATSENTRY']._loaded_options = None
+  _globals['_COMBINEDTFDATASTATS_TFDATASTATSENTRY']._serialized_options = b'8\001'
+  _globals['_ITERATORSTAT']._serialized_start=67
+  _globals['_ITERATORSTAT']._serialized_end=199
+  _globals['_ITERATORMETADATA']._serialized_start=202
+  _globals['_ITERATORMETADATA']._serialized_end=416
+  _globals['_ITERATORMETADATA_PARAMSENTRY']._serialized_start=371
+  _globals['_ITERATORMETADATA_PARAMSENTRY']._serialized_end=416
+  _globals['_INPUTPIPELINESTAT']._serialized_start=419
+  _globals['_INPUTPIPELINESTAT']._serialized_end=682
+  _globals['_INPUTPIPELINESTAT_ITERATORSTATSENTRY']._serialized_start=595
+  _globals['_INPUTPIPELINESTAT_ITERATORSTATSENTRY']._serialized_end=682
+  _globals['_INPUTPIPELINEMETADATA']._serialized_start=685
+  _globals['_INPUTPIPELINEMETADATA']._serialized_end=859
+  _globals['_INPUTPIPELINEMETADATA_INPUTPIPELINETYPE']._serialized_start=812
+  _globals['_INPUTPIPELINEMETADATA_INPUTPIPELINETYPE']._serialized_end=853
+  _globals['_INPUTPIPELINESTATS']._serialized_start=862
+  _globals['_INPUTPIPELINESTATS']._serialized_end=1095
+  _globals['_TFDATASTATS']._serialized_start=1098
+  _globals['_TFDATASTATS']._serialized_end=1465
+  _globals['_TFDATASTATS_ITERATORMETADATAENTRY']._serialized_start=1275
+  _globals['_TFDATASTATS_ITERATORMETADATAENTRY']._serialized_end=1369
+  _globals['_TFDATASTATS_INPUTPIPELINESENTRY']._serialized_start=1371
+  _globals['_TFDATASTATS_INPUTPIPELINESENTRY']._serialized_end=1465
+  _globals['_TFDATABOTTLENECKANALYSIS']._serialized_start=1468
+  _globals['_TFDATABOTTLENECKANALYSIS']._serialized_end=1656
+  _globals['_COMBINEDTFDATASTATS']._serialized_start=1659
+  _globals['_COMBINEDTFDATASTATS']._serialized_end=1965
+  _globals['_COMBINEDTFDATASTATS_TFDATASTATSENTRY']._serialized_start=1881
+  _globals['_COMBINEDTFDATASTATS_TFDATASTATSENTRY']._serialized_end=1965
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/tf_function_pb2.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/tf_function.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/tf_function.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'plugin/xprof/protobuf/tf_function.proto\x12\x13tensorflow.profiler\"8\n\x11TfFunctionMetrics\x12\r\n\x05\x63ount\x18\x01 \x01(\x04\x12\x14\n\x0cself_time_ps\x18\x02 \x01(\x04\"\x9b\x02\n\nTfFunction\x12=\n\x07metrics\x18\x01 \x03(\x0b\x32,.tensorflow.profiler.TfFunction.MetricsEntry\x12\x1b\n\x13total_tracing_count\x18\x02 \x01(\x03\x12\x39\n\x08\x63ompiler\x18\x03 \x01(\x0e\x32\'.tensorflow.profiler.TfFunctionCompiler\x12\x1e\n\x16\x65xpensive_call_percent\x18\x04 \x01(\x01\x1aV\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32&.tensorflow.profiler.TfFunctionMetrics:\x02\x38\x01\"\xad\x01\n\x0cTfFunctionDb\x12H\n\x0ctf_functions\x18\x01 \x03(\x0b\x32\x32.tensorflow.profiler.TfFunctionDb.TfFunctionsEntry\x1aS\n\x10TfFunctionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12.\n\x05value\x18\x02 \x01(\x0b\x32\x1f.tensorflow.profiler.TfFunction:\x02\x38\x01*t\n\x17TfFunctionExecutionMode\x12\x10\n\x0cINVALID_MODE\x10\x00\x12\x0e\n\nEAGER_MODE\x10\x01\x12\x0f\n\x0bTRACED_MODE\x10\x02\x12\x13\n\x0fNOT_TRACED_MODE\x10\x03\x12\x11\n\rCONCRETE_MODE\x10\x04*w\n\x12TfFunctionCompiler\x12\x14\n\x10INVALID_COMPILER\x10\x00\x12\x12\n\x0eOTHER_COMPILER\x10\x01\x12\x12\n\x0eMIXED_COMPILER\x10\x02\x12\x10\n\x0cXLA_COMPILER\x10\x03\x12\x11\n\rMLIR_COMPILER\x10\x04\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.tf_function_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_TFFUNCTION_METRICSENTRY']._loaded_options = None
+  _globals['_TFFUNCTION_METRICSENTRY']._serialized_options = b'8\001'
+  _globals['_TFFUNCTIONDB_TFFUNCTIONSENTRY']._loaded_options = None
+  _globals['_TFFUNCTIONDB_TFFUNCTIONSENTRY']._serialized_options = b'8\001'
+  _globals['_TFFUNCTIONEXECUTIONMODE']._serialized_start=584
+  _globals['_TFFUNCTIONEXECUTIONMODE']._serialized_end=700
+  _globals['_TFFUNCTIONCOMPILER']._serialized_start=702
+  _globals['_TFFUNCTIONCOMPILER']._serialized_end=821
+  _globals['_TFFUNCTIONMETRICS']._serialized_start=64
+  _globals['_TFFUNCTIONMETRICS']._serialized_end=120
+  _globals['_TFFUNCTION']._serialized_start=123
+  _globals['_TFFUNCTION']._serialized_end=406
+  _globals['_TFFUNCTION_METRICSENTRY']._serialized_start=320
+  _globals['_TFFUNCTION_METRICSENTRY']._serialized_end=406
+  _globals['_TFFUNCTIONDB']._serialized_start=409
+  _globals['_TFFUNCTIONDB']._serialized_end=582
+  _globals['_TFFUNCTIONDB_TFFUNCTIONSENTRY']._serialized_start=499
+  _globals['_TFFUNCTIONDB_TFFUNCTIONSENTRY']._serialized_end=582
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/tf_stats_pb2.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/tf_stats.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/tf_stats.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$plugin/xprof/protobuf/tf_stats.proto\x12\x13tensorflow.profiler\"\xa7\x01\n\x0fTfStatsDatabase\x12\x34\n\twith_idle\x18\x04 \x01(\x0b\x32!.tensorflow.profiler.TfStatsTable\x12\x37\n\x0cwithout_idle\x18\x05 \x01(\x0b\x32!.tensorflow.profiler.TfStatsTable\x12\x13\n\x0b\x64\x65vice_type\x18\x06 \x01(\tJ\x04\x08\x01\x10\x02J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04\"\x83\x01\n\x0cTfStatsTable\x12;\n\x0ftf_stats_record\x18\x01 \x03(\x0b\x32\".tensorflow.profiler.TfStatsRecord\x12\x19\n\x11host_tf_pprof_key\x18\x02 \x01(\t\x12\x1b\n\x13\x64\x65vice_tf_pprof_key\x18\x03 \x01(\t\"\xd8\x07\n\rTfStatsRecord\x12\x0c\n\x04rank\x18\x01 \x01(\x04\x12\x16\n\x0ehost_or_device\x18\x02 \x01(\t\x12\x0f\n\x07op_type\x18\x03 \x01(\t\x12\x0f\n\x07op_name\x18\x04 \x01(\t\x12\x13\n\x0boccurrences\x18\x05 \x01(\x03\x12\x18\n\x10total_time_in_us\x18\x06 \x01(\x01\x12\x16\n\x0e\x61vg_time_in_us\x18\x07 \x01(\x01\x12\x1d\n\x15total_self_time_in_us\x18\x08 \x01(\x01\x12\x1b\n\x13\x61vg_self_time_in_us\x18\t \x01(\x01\x12*\n\"device_total_self_time_as_fraction\x18\n \x01(\x01\x12\x35\n-device_cumulative_total_self_time_as_fraction\x18\x0b \x01(\x01\x12(\n host_total_self_time_as_fraction\x18\x0c \x01(\x01\x12\x33\n+host_cumulative_total_self_time_as_fraction\x18\r \x01(\x01\x12\x1a\n\x12measured_flop_rate\x18\x0e \x01(\x01\x12\x17\n\x0fmodel_flop_rate\x18\x14 \x01(\x01\x12\x1a\n\x12measured_memory_bw\x18\x0f \x01(\x01\x12\x1d\n\x15operational_intensity\x18\x10 \x01(\x01\x12\x10\n\x08\x62ound_by\x18\x11 \x01(\t\x12\x10\n\x08is_eager\x18\x12 \x01(\x08\x12\"\n\x1agpu_tensorcore_utilization\x18\x13 \x01(\x01\x12\x0e\n\x06hbm_bw\x18\x15 \x01(\x01\x12\x14\n\x0c\x63mem_read_bw\x18\x16 \x01(\x01\x12\x15\n\rcmem_write_bw\x18\x17 \x01(\x01\x12\x14\n\x0cvmem_read_bw\x18\x18 \x01(\x01\x12\x15\n\rvmem_write_bw\x18\x19 \x01(\x01\x12!\n\x19hbm_operational_intensity\x18\x1a \x01(\x01\x12\'\n\x1f\x63mem_read_operational_intensity\x18\x1b \x01(\x01\x12(\n cmem_write_operational_intensity\x18\x1c \x01(\x01\x12\'\n\x1fvmem_read_operational_intensity\x18\x1d \x01(\x01\x12(\n vmem_write_operational_intensity\x18\x1e \x01(\x01\x12(\n bottleneck_operational_intensity\x18\x1f \x01(\x01\x12\r\n\x05\x66lops\x18  \x01(\x04\x12\x16\n\x0e\x62ytes_accessed\x18! \x01(\x04\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.tf_stats_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_TFSTATSDATABASE']._serialized_start=62
+  _globals['_TFSTATSDATABASE']._serialized_end=229
+  _globals['_TFSTATSTABLE']._serialized_start=232
+  _globals['_TFSTATSTABLE']._serialized_end=363
+  _globals['_TFSTATSRECORD']._serialized_start=366
+  _globals['_TFSTATSRECORD']._serialized_end=1350
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/tfstreamz_pb2.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/tfstreamz.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/tfstreamz.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%plugin/xprof/protobuf/tfstreamz.proto\x12\x1dtensorflow.profiler.tfstreamz\"4\n\x0fPercentilePoint\x12\x12\n\npercentile\x18\x01 \x01(\x01\x12\r\n\x05value\x18\x02 \x01(\x01\"\xc3\x02\n\x0bPercentiles\x12\x45\n\x0funit_of_measure\x18\x01 \x01(\x0e\x32,.tensorflow.profiler.tfstreamz.UnitOfMeasure\x12\x14\n\x0cstart_nstime\x18\x02 \x01(\x04\x12\x12\n\nend_nstime\x18\x03 \x01(\x04\x12\x11\n\tmin_value\x18\x04 \x01(\x01\x12\x11\n\tmax_value\x18\x05 \x01(\x01\x12\x0c\n\x04mean\x18\x06 \x01(\x01\x12\x0e\n\x06stddev\x18\x07 \x01(\x01\x12\x13\n\x0bnum_samples\x18\x08 \x01(\x04\x12\x15\n\rtotal_samples\x18\t \x01(\x04\x12\x13\n\x0b\x61\x63\x63umulator\x18\n \x01(\x01\x12>\n\x06points\x18\x0b \x03(\x0b\x32..tensorflow.profiler.tfstreamz.PercentilePoint*0\n\rUnitOfMeasure\x12\n\n\x06NUMBER\x10\x00\x12\x08\n\x04TIME\x10\x01\x12\t\n\x05\x42YTES\x10\x02\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.tfstreamz_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_UNITOFMEASURE']._serialized_start=452
+  _globals['_UNITOFMEASURE']._serialized_end=500
+  _globals['_PERCENTILEPOINT']._serialized_start=72
+  _globals['_PERCENTILEPOINT']._serialized_end=124
+  _globals['_PERCENTILES']._serialized_start=127
+  _globals['_PERCENTILES']._serialized_end=450
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/topology_pb2.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/topology.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/topology.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$plugin/xprof/protobuf/topology.proto\x12\x13tensorflow.profiler\"4\n\x11TopologyDimension\x12\t\n\x01x\x18\x01 \x01(\x05\x12\t\n\x01y\x18\x02 \x01(\x05\x12\t\n\x01z\x18\x03 \x01(\x05\"\x8d\x01\n\x10TopologyLocation\x12\t\n\x01x\x18\x01 \x01(\x05\x12\t\n\x01y\x18\x02 \x01(\x05\x12\t\n\x01z\x18\x03 \x01(\x05\x12\x0e\n\x06host_x\x18\x04 \x01(\x05\x12\x0e\n\x06host_y\x18\x05 \x01(\x05\x12\x0e\n\x06host_z\x18\x06 \x01(\x05\x12\x15\n\rindex_on_host\x18\x07 \x01(\x05\x12\x11\n\tglobal_id\x18\x08 \x01(\x05\"\xcc\x01\n\x08Topology\x12\x45\n\x15\x63hips_per_host_bounds\x18\x01 \x01(\x0b\x32&.tensorflow.profiler.TopologyDimension\x12;\n\x0bhost_bounds\x18\x02 \x01(\x0b\x32&.tensorflow.profiler.TopologyDimension\x12<\n\rmesh_location\x18\x03 \x03(\x0b\x32%.tensorflow.profiler.TopologyLocation\"\x8b\x04\n\x0fLogicalTopology\x12\x41\n\x06slices\x18\x01 \x03(\x0b\x32\x31.tensorflow.profiler.LogicalTopology.LogicalSlice\x1aQ\n\rLogicalDevice\x12\x11\n\tglobal_id\x18\x01 \x01(\x05\x12\x16\n\x0eslice_local_id\x18\x02 \x01(\x05\x12\x15\n\rhost_local_id\x18\x03 \x01(\x05\x1a=\n\x12HostNetworkAddress\x12\x0f\n\x07\x61\x64\x64ress\x18\x01 \x01(\t\x12\x16\n\x0einterface_name\x18\x02 \x01(\t\x1a\xbe\x01\n\x0bLogicalHost\x12\x16\n\x0eslice_local_id\x18\x01 \x01(\x05\x12R\n\x11network_addresses\x18\x02 \x03(\x0b\x32\x37.tensorflow.profiler.LogicalTopology.HostNetworkAddress\x12\x43\n\x07\x64\x65vices\x18\x03 \x03(\x0b\x32\x32.tensorflow.profiler.LogicalTopology.LogicalDevice\x1a\x62\n\x0cLogicalSlice\x12\x11\n\tglobal_id\x18\x01 \x01(\x05\x12?\n\x05hosts\x18\x02 \x03(\x0b\x32\x30.tensorflow.profiler.LogicalTopology.LogicalHostb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.topology_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_TOPOLOGYDIMENSION']._serialized_start=61
+  _globals['_TOPOLOGYDIMENSION']._serialized_end=113
+  _globals['_TOPOLOGYLOCATION']._serialized_start=116
+  _globals['_TOPOLOGYLOCATION']._serialized_end=257
+  _globals['_TOPOLOGY']._serialized_start=260
+  _globals['_TOPOLOGY']._serialized_end=464
+  _globals['_LOGICALTOPOLOGY']._serialized_start=467
+  _globals['_LOGICALTOPOLOGY']._serialized_end=990
+  _globals['_LOGICALTOPOLOGY_LOGICALDEVICE']._serialized_start=553
+  _globals['_LOGICALTOPOLOGY_LOGICALDEVICE']._serialized_end=634
+  _globals['_LOGICALTOPOLOGY_HOSTNETWORKADDRESS']._serialized_start=636
+  _globals['_LOGICALTOPOLOGY_HOSTNETWORKADDRESS']._serialized_end=697
+  _globals['_LOGICALTOPOLOGY_LOGICALHOST']._serialized_start=700
+  _globals['_LOGICALTOPOLOGY_LOGICALHOST']._serialized_end=890
+  _globals['_LOGICALTOPOLOGY_LOGICALSLICE']._serialized_start=892
+  _globals['_LOGICALTOPOLOGY_LOGICALSLICE']._serialized_end=990
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/tpu_input_pipeline_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/tpu_input_pipeline.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/tpu_input_pipeline.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import input_pipeline_pb2 as plugin_dot_xprof_dot_protobuf_dot_input__pipeline__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.plugin/xprof/protobuf/tpu_input_pipeline.proto\x12\x13tensorflow.profiler\x1a*plugin/xprof/protobuf/input_pipeline.proto\"\xee\x04\n\x11PerTpuStepDetails\x12\x13\n\x0bstep_number\x18\x01 \x01(\x05\x12\x1a\n\x12tc_compute_time_ms\x18\r \x01(\x01\x12\x17\n\x0ftc_idle_time_ms\x18\x0e \x01(\x01\x12\x1a\n\x12tc_outfeed_time_ms\x18\x0f \x01(\x01\x12\x19\n\x11tc_infeed_time_ms\x18\x03 \x01(\x01\x12\x1e\n\x16infeed_percent_average\x18\x04 \x01(\x01\x12\x1e\n\x16infeed_percent_minimum\x18\x05 \x01(\x01\x12\x1e\n\x16infeed_percent_maximum\x18\x06 \x01(\x01\x12\x1e\n\x16\x63oreid_max_infeed_time\x18\x07 \x01(\r\x12!\n\x19max_infeed_time_core_name\x18\x19 \x01(\t\x12\"\n\x1a\x61ll_reduce_compute_time_ms\x18\x0b \x01(\x01\x12\x1f\n\x17\x61ll_reduce_sync_time_ms\x18\x0c \x01(\x01\x12\x1c\n\x14scv0_compute_time_ms\x18\x10 \x01(\x01\x12\x1b\n\x13scv0_infeed_time_ms\x18\x11 \x01(\x01\x12\x18\n\x10host_transfer_ms\x18\x12 \x01(\x01\x12\x1a\n\x12sc_compute_time_ms\x18\x14 \x01(\x01\x12\x17\n\x0fsc_idle_time_ms\x18\x15 \x01(\x01\x12\x1a\n\x12sc_outfeed_time_ms\x18\x16 \x01(\x01\x12\x19\n\x11sc_infeed_time_ms\x18\x17 \x01(\x01\x12\x17\n\x0fsc_step_time_ms\x18\x18 \x01(\x01J\x04\x08\x02\x10\x03J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"\xad\x04\n\x14TpuStepTimeBreakdown\x12?\n\x15tc_compute_ms_summary\x18\x01 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12\x41\n\x17scv0_compute_ms_summary\x18\x02 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12>\n\x14tc_infeed_ms_summary\x18\x03 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12?\n\x15tc_outfeed_ms_summary\x18\x06 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12@\n\x16scv0_infeed_ms_summary\x18\x04 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12<\n\x12tc_idle_ms_summary\x18\x05 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12\x42\n\x18host_transfer_ms_summary\x18\x07 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12L\n\x18sparse_core_step_summary\x18\x08 \x01(\x0b\x32*.tensorflow.profiler.SparseCoreStepSummary\"\xda\x02\n\x15SparseCoreStepSummary\x12?\n\x15sc_compute_ms_summary\x18\x01 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12>\n\x14sc_infeed_ms_summary\x18\x02 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12?\n\x15sc_outfeed_ms_summary\x18\x03 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12<\n\x12sc_idle_ms_summary\x18\x04 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\x12\x41\n\x17sc_step_time_ms_summary\x18\x05 \x01(\x0b\x32 .tensorflow.profiler.StepSummary\"\x99\x03\n\x15TpuBottleneckAnalysis\x12\x15\n\rinput_percent\x18\x0b \x01(\x01\x12\x1c\n\x14input_classification\x18\x01 \x01(\t\x12\x17\n\x0finput_statement\x18\x02 \x01(\t\x12\x16\n\x0eoutput_percent\x18\x0c \x01(\x01\x12\x1d\n\x15output_classification\x18\t \x01(\t\x12\x18\n\x10output_statement\x18\n \x01(\t\x12\x17\n\x0ftc_idle_percent\x18\r \x01(\x01\x12\x1e\n\x16tc_idle_classification\x18\x03 \x01(\t\x12\x19\n\x11tc_idle_statement\x18\x04 \x01(\t\x12\x1b\n\x13scv0_classification\x18\x05 \x01(\t\x12\x16\n\x0escv0_statement\x18\x06 \x01(\t\x12!\n\x19\x61ll_reduce_classification\x18\x07 \x01(\t\x12\x1c\n\x14\x61ll_reduce_statement\x18\x08 \x01(\t\x12\x17\n\x0f\x63ompute_percent\x18\x0e \x01(\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.tpu_input_pipeline_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_PERTPUSTEPDETAILS']._serialized_start=116
+  _globals['_PERTPUSTEPDETAILS']._serialized_end=738
+  _globals['_TPUSTEPTIMEBREAKDOWN']._serialized_start=741
+  _globals['_TPUSTEPTIMEBREAKDOWN']._serialized_end=1298
+  _globals['_SPARSECORESTEPSUMMARY']._serialized_start=1301
+  _globals['_SPARSECORESTEPSUMMARY']._serialized_end=1647
+  _globals['_TPUBOTTLENECKANALYSIS']._serialized_start=1650
+  _globals['_TPUBOTTLENECKANALYSIS']._serialized_end=2059
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/trace_events_old_pb2.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/trace_events_old.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/trace_events_old.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,plugin/xprof/protobuf/trace_events_old.proto\x12\x05xprof\"\x9b\x01\n\x05Trace\x12*\n\x07\x64\x65vices\x18\x01 \x03(\x0b\x32\x19.xprof.Trace.DevicesEntry\x12\'\n\x0ctrace_events\x18\x04 \x03(\x0b\x32\x11.xprof.TraceEvent\x1a=\n\x0c\x44\x65vicesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.xprof.Device:\x02\x38\x01\"\x9d\x01\n\x06\x44\x65vice\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\tdevice_id\x18\x02 \x01(\x04\x12/\n\tresources\x18\x03 \x03(\x0b\x32\x1c.xprof.Device.ResourcesEntry\x1a\x41\n\x0eResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12\x1e\n\x05value\x18\x02 \x01(\x0b\x32\x0f.xprof.Resource:\x02\x38\x01\"-\n\x08Resource\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0bresource_id\x18\x02 \x01(\x04\"\xc5\x01\n\nTraceEvent\x12\x11\n\tdevice_id\x18\x01 \x01(\x04\x12\x13\n\x0bresource_id\x18\x02 \x01(\x04\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x14\n\x0ctimestamp_ps\x18\t \x01(\x04\x12\x13\n\x0b\x64uration_ps\x18\n \x01(\x04\x12)\n\x04\x61rgs\x18\x0b \x03(\x0b\x32\x1b.xprof.TraceEvent.ArgsEntry\x1a+\n\tArgsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.trace_events_old_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_TRACE_DEVICESENTRY']._loaded_options = None
+  _globals['_TRACE_DEVICESENTRY']._serialized_options = b'8\001'
+  _globals['_DEVICE_RESOURCESENTRY']._loaded_options = None
+  _globals['_DEVICE_RESOURCESENTRY']._serialized_options = b'8\001'
+  _globals['_TRACEEVENT_ARGSENTRY']._loaded_options = None
+  _globals['_TRACEEVENT_ARGSENTRY']._serialized_options = b'8\001'
+  _globals['_TRACE']._serialized_start=56
+  _globals['_TRACE']._serialized_end=211
+  _globals['_TRACE_DEVICESENTRY']._serialized_start=150
+  _globals['_TRACE_DEVICESENTRY']._serialized_end=211
+  _globals['_DEVICE']._serialized_start=214
+  _globals['_DEVICE']._serialized_end=371
+  _globals['_DEVICE_RESOURCESENTRY']._serialized_start=306
+  _globals['_DEVICE_RESOURCESENTRY']._serialized_end=371
+  _globals['_RESOURCE']._serialized_start=373
+  _globals['_RESOURCE']._serialized_end=418
+  _globals['_TRACEEVENT']._serialized_start=421
+  _globals['_TRACEEVENT']._serialized_end=618
+  _globals['_TRACEEVENT_ARGSENTRY']._serialized_start=575
+  _globals['_TRACEEVENT_ARGSENTRY']._serialized_end=618
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/trace_events_pb2.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/trace_events.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/trace_events.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import task_pb2 as plugin_dot_xprof_dot_protobuf_dot_task__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(plugin/xprof/protobuf/trace_events.proto\x12\x13tensorflow.profiler\x1a plugin/xprof/protobuf/task.proto\"\xd2\x03\n\x05Trace\x12\x38\n\x07\x64\x65vices\x18\x01 \x03(\x0b\x32\'.tensorflow.profiler.Trace.DevicesEntry\x12\x34\n\x05tasks\x18\x06 \x03(\x0b\x32%.tensorflow.profiler.Trace.TasksEntry\x12\x18\n\x10min_timestamp_ps\x18\x04 \x01(\x04\x12\x18\n\x10max_timestamp_ps\x18\x05 \x01(\x04\x12\x12\n\nnum_events\x18\x07 \x01(\x04\x12=\n\nname_table\x18\x08 \x03(\x0b\x32).tensorflow.profiler.Trace.NameTableEntry\x1aK\n\x0c\x44\x65vicesEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x1b.tensorflow.profiler.Device:\x02\x38\x01\x1aG\n\nTasksEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.tensorflow.profiler.Task:\x02\x38\x01\x1a\x30\n\x0eNameTableEntry\x12\x0b\n\x03key\x18\x01 \x01(\x06\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04\"\xbf\x01\n\x06\x44\x65vice\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\tdevice_id\x18\x02 \x01(\r\x12=\n\tresources\x18\x03 \x03(\x0b\x32*.tensorflow.profiler.Device.ResourcesEntry\x1aO\n\x0eResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x1d.tensorflow.profiler.Resource:\x02\x38\x01J\x04\x08\x04\x10\x05\"A\n\x08Resource\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0bresource_id\x18\x02 \x01(\x04\x12\x12\n\nnum_events\x18\x03 \x01(\r\"\xba\x04\n\nTraceEvent\x12\x37\n\x04type\x18\x0e \x01(\x0e\x32).tensorflow.profiler.TraceEvent.EventType\x12\x11\n\tdevice_id\x18\x01 \x01(\r\x12\x13\n\x0bresource_id\x18\x02 \x01(\x04\x12\x0e\n\x04name\x18\x03 \x01(\tH\x00\x12\x12\n\x08name_ref\x18\x0c \x01(\x06H\x00\x12\x14\n\x08group_id\x18\x05 \x01(\x03:\x02-1\x12\x14\n\x0ctimestamp_ps\x18\x06 \x01(\x04\x12\x13\n\x0b\x64uration_ps\x18\x07 \x01(\x04\x12\x10\n\x08raw_data\x18\x08 \x01(\x0c\x12\x0f\n\x07\x66low_id\x18\t \x01(\x04\x12\x46\n\x0f\x66low_entry_type\x18\n \x01(\x0e\x32-.tensorflow.profiler.TraceEvent.FlowEntryType\x12\x15\n\rflow_category\x18\x0b \x01(\r\x12\x0e\n\x06serial\x18\r \x01(\r\"t\n\tEventType\x12\x1a\n\x16\x45VENT_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13\x45VENT_TYPE_COMPLETE\x10\x01\x12\x14\n\x10\x45VENT_TYPE_ASYNC\x10\x03\x12\x16\n\x12\x45VENT_TYPE_COUNTER\x10\x04\"\x04\x08\x02\x10\x02\"J\n\rFlowEntryType\x12\r\n\tFLOW_NONE\x10\x00\x12\x0e\n\nFLOW_START\x10\x01\x12\x0c\n\x08\x46LOW_MID\x10\x02\x12\x0c\n\x08\x46LOW_END\x10\x03\x42\x0c\n\nname_oneofJ\x04\x08\x04\x10\x05\x42\x03\xf8\x01\x01')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.trace_events_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\370\001\001'
+  _globals['_TRACE_DEVICESENTRY']._loaded_options = None
+  _globals['_TRACE_DEVICESENTRY']._serialized_options = b'8\001'
+  _globals['_TRACE_TASKSENTRY']._loaded_options = None
+  _globals['_TRACE_TASKSENTRY']._serialized_options = b'8\001'
+  _globals['_TRACE_NAMETABLEENTRY']._loaded_options = None
+  _globals['_TRACE_NAMETABLEENTRY']._serialized_options = b'8\001'
+  _globals['_DEVICE_RESOURCESENTRY']._loaded_options = None
+  _globals['_DEVICE_RESOURCESENTRY']._serialized_options = b'8\001'
+  _globals['_TRACE']._serialized_start=100
+  _globals['_TRACE']._serialized_end=566
+  _globals['_TRACE_DEVICESENTRY']._serialized_start=356
+  _globals['_TRACE_DEVICESENTRY']._serialized_end=431
+  _globals['_TRACE_TASKSENTRY']._serialized_start=433
+  _globals['_TRACE_TASKSENTRY']._serialized_end=504
+  _globals['_TRACE_NAMETABLEENTRY']._serialized_start=506
+  _globals['_TRACE_NAMETABLEENTRY']._serialized_end=554
+  _globals['_DEVICE']._serialized_start=569
+  _globals['_DEVICE']._serialized_end=760
+  _globals['_DEVICE_RESOURCESENTRY']._serialized_start=675
+  _globals['_DEVICE_RESOURCESENTRY']._serialized_end=754
+  _globals['_RESOURCE']._serialized_start=762
+  _globals['_RESOURCE']._serialized_end=827
+  _globals['_TRACEEVENT']._serialized_start=830
+  _globals['_TRACEEVENT']._serialized_end=1400
+  _globals['_TRACEEVENT_EVENTTYPE']._serialized_start=1188
+  _globals['_TRACEEVENT_EVENTTYPE']._serialized_end=1304
+  _globals['_TRACEEVENT_FLOWENTRYTYPE']._serialized_start=1306
+  _globals['_TRACEEVENT_FLOWENTRYTYPE']._serialized_end=1380
+# @@protoc_insertion_point(module_scope)