
    =VhIM                        d Z ddlmZ ddlmZ ddlmZ ddlm	Z
  e
j                         ZddlmZ ddlmZ ddlmZ  ej                         j+                  d	      Z ej.                  e e               ej2                  ed
 e              ej4                  dk(  rde_        de_        dej<                  d   _        dej<                  d   _        dej<                  d   _        dej<                  d   _        dej<                  d   _        dej<                  d   _        de_        de_        de j<                  d   _        de j<                  d   _        de!_"        de!_#        de$_"        de$_#        de%_"        de%_#        de_"        de_#        de&_"        de&_#        de'_"        de'_#        d e(_"        d!e(_#        d"e)_"        d#e)_#        d$e_"        d!e_#        d%e*_"        d&e*_#        d'e+_"        d&e+_#        d(e,_"        d)e,_#        d*e-_"        d)e-_#        d+e._"        d,e._#        d-e/_"        d.e/_#        d/e0_"        d0e0_#        d1e1_"        d.e1_#        d2e2_"        d3e2_#        d4e3_"        d5e3_#        d6e4_"        d7e4_#        d8e5_"        d9e5_#        d:e6_"        d;e6_#        d<e _"        d;e _#        d=e7_"        d>e7_#        d?e8_"        d>e8_#        d@e9_"        dAe9_#        dBe:_"        dAe:_#        dCe;_"        dAe;_#        dDe<_"        dEe<_#        dFe=_"        dGe=_#        dHe>_"        dGe>_#        dIe?_"        dJe?_#        dKe@_"        dLe@_#        dMeA_"        dNeA_#        dOeB_"        dPeB_#        dQeC_"        dReC_#        dSeD_"        dTeD_#        yy)UzGenerated protocol buffer code.    )builder)
descriptor)descriptor_pool)symbol_database)any_pb2)metrics_pb2)xla_data_pb2sn!  
xla/service/hlo.protoxlagoogle/protobuf/any.protoxla/service/metrics.protoxla/xla_data.proto"
HloInstructionProto
name (	
opcode (	
shape (2.xla.ShapeProto!
metadata (2.xla.OpMetadata"
literal (2.xla.LiteralProto
parameter_number	 (
fusion_kind (	
tuple_index (

dimensions (
window (2.xla.WindowG
convolution_dimension_numbers (2 .xla.ConvolutionDimensionNumbers
feature_group_count2 (
batch_group_count: (B
slice_dimensions (2(.xla.HloInstructionProto.SliceDimensions
exponent_bits (
mantissa_bits (
dynamic_slice_sizes (*
padding_config (2.xla.PaddingConfig
outfeed_config (-
distribution (2.xla.RandomDistribution
epsilon (
feature_index (

channel_id (
infeed_config (
custom_call_target (	&
outfeed_shape (2.xla.ShapeProto7
dot_dimension_numbers (2.xla.DotDimensionNumbersD
ragged_dot_dimension_numbersZ (2.xla.RaggedDotDimensionNumbers
fft_type (2.xla.FftType

fft_length  (
comparison_direction? (	=
gather_dimension_numbers! (2.xla.GatherDimensionNumbers
gather_slice_sizes" (

id# (
operand_ids$ (
control_predecessor_ids% (
called_computation_ids& (!
sharding( (2.xla.OpSharding
backend_config+ (-
replica_groups1 (2.xla.ReplicaGroupB
all_reduce_id- (B
use_global_device_idsG (
is_host_transfer/ (
	is_stable< (?
scatter_dimension_numbers0 (2.xla.ScatterDimensionNumbers.
precision_config3 (2.xla.PrecisionConfig.
source_target_pairs4 (2.xla.SourceTarget.
domain_entry_sharding6 (2.xla.OpSharding-
domain_exit_sharding7 (2.xla.OpSharding
constrain_layout8 (3
operand_shapes_with_layout9 (2.xla.ShapeProto=
triangular_solve_options; (2.xla.TriangularSolveOptions.
cholesky_options> (2.xla.CholeskyOptions8
parameter_replication= (2.xla.ParameterReplication#
custom_call_has_side_effectA (;
output_operand_aliasingJ (2.xla.OutputOperandAliasing5
custom_call_scheduleL (2.xla.CustomCallSchedule
deltaB (
indices_are_sortedC (4
frontend_attributesD (2.xla.FrontendAttributes
unique_indicesE (+
rng_algorithmF (2.xla.RandomAlgorithm
comparison_typeH (	%
is_cross_program_prefetchI (B&
cross_program_prefetch_indexP (H &
padding_typeK (2.xla.PaddingType:
custom_call_api_versionM (2.xla.CustomCallApiVersion
async_execution_threadO (		
kQ (
largestU (*
statistics_vizR (2.xla.StatisticsViz-
dot_sparsityV (2.xla.SparsityDescriptor>
collective_device_listW (2.xla.CollectiveDeviceListProto/
original_valueX (2.xla.OriginalValueProto
is_compositeY (,
result_accuracy[ (2.xla.ResultAccuracy?
SliceDimensions
start (
limit (
stride (B'
%optional_cross_program_prefetch_indexJ
JJJJJ,-J56J./J)*J*+J@AJNOJSTJTURparameter_nameRfused_instructions_computationRoperand_namesRcontrol_predecessor_namesRcalled_computation_namesRreplica_group_idsRcustom_call_opaqueRall_reduce_barrier"
HloComputationProto
name (	.
instructions (2.xla.HloInstructionProto-
program_shape (2.xla.ProgramShapeProto

id (
root_id (
is_fusion_computation (
execution_thread (	JR	root_name"
HloScheduleProto7
	sequences (2$.xla.HloScheduleProto.SequencesEntry.
InstructionSequence
instruction_ids ([
SequencesEntry
key (8
value (2).xla.HloScheduleProto.InstructionSequence:8"
HloInputOutputAliasProto>
entries (2-.xla.HloInputOutputAliasProto.AliasEntryProto
AliasEntryProto
output_shape_index (
parameter_number (
parameter_shape_index (
kind (2	.xla.Kind"
HloBufferDonorProto?
entries (2..xla.HloBufferDonorProto.BufferDonorEntryProtoP
BufferDonorEntryProto
parameter_number (
parameter_shape_index ("H
CrossProgramPrefetch
	parameter (
index (
offset ("
StackFrameIndexProto

file_names (	
function_names (	>
file_locations (2&.xla.StackFrameIndexProto.FileLocation:
stack_frames (2$.xla.StackFrameIndexProto.StackFrame\
FileLocation
file_name_id (
function_name_id (
line (
column (?

StackFrame
file_location_id (
parent_frame_id ("

HloModuleProto
name (	
entry_computation_name (	
entry_computation_id (.
computations (2.xla.HloComputationProto2
host_program_shape (2.xla.ProgramShapeProto

id ('
schedule (2.xla.HloScheduleProto9
input_output_alias (2.xla.HloInputOutputAliasProto.
buffer_donor (2.xla.HloBufferDonorProto;
cross_program_prefetches
 (2.xla.CrossProgramPrefetch

is_dynamic (-
spmd_output_sharding (2.xla.OpSharding2
spmd_parameters_shardings (2.xla.OpSharding"
use_auto_spmd_partitioning (5
profile_info (2.xla.HloModuleProto.ProfileInfo5
device_assignment (2.xla.DeviceAssignmentProto4
stack_frame_index (2.xla.StackFrameIndexProto4
frontend_attributes (2.xla.FrontendAttributes
ProfileInfo5
profile_type (2.xla.HloModuleProto.ProfileType
relative_speedup (*
profile_source (2.xla.ProfileSource0
compilation_event (2.xla.CompilationEvent
fingerprint (	R
profile_generation_strategy (2-.xla.HloModuleProto.ProfileGenerationStrategy"`
ProfileType
INVALID 
FLAG

FUSION

LAYOUT
DOT
FLAGNET
SHARDING"
ProfileGenerationStrategy'
#PROFILE_GENERATION_STRATEGY_UNKNOWN "
PROFILE_GENERATION_STRATEGY_GA%
!PROFILE_GENERATION_STRATEGY_FANTA#
PROFILE_GENERATION_STRATEGY_CFOJ	
Rdynamic_parameter_binding"
LogicalBufferProto

id (
size (4

defined_at (2 .xla.LogicalBufferProto.Location
color ([
Location
instruction_name (	B
instruction_id (
shape_index (J"
BufferAllocationProto
index (
size (
is_thread_local (
is_tuple (&
is_entry_computation_parameter (
is_constant (
parameter_number (
parameter_shape_index
 (
maybe_live_out (
color (5
assigned	 (2#.xla.BufferAllocationProto.AssignedC
Assigned
logical_buffer_id (
offset (
size ("
HeapSimulatorTrace-
events (2.xla.HeapSimulatorTrace.Event
whole_module_simulation (
buffer_allocation_index (
Event0
kind (2".xla.HeapSimulatorTrace.Event.Kind
	buffer_id (
computation_name (	
instruction_name (	
share_with_canonical_id ("+
Kind	
ALLOC 
FREE

SHARE_WITH"M
HloModuleGroupProto
name (	(
hlo_modules (2.xla.HloModuleProto"
BufferAssignmentProto0
logical_buffers (2.xla.LogicalBufferProto>
buffer_aliases (2&.xla.BufferAssignmentProto.BufferAlias6
buffer_allocations (2.xla.BufferAllocationProto6
heap_simulator_traces (2.xla.HeapSimulatorTrace[
BufferAlias
source_buffer_id (2
location (2 .xla.LogicalBufferProto.Location"~
HloProto'

hlo_module (2.xla.HloModuleProto5
buffer_assignment (2.xla.BufferAssignmentProtoJRhlo_ordering"
HloSnapshot
hlo (2.xla.HloProto$
	arguments (2.xla.LiteralProto!
result (2.xla.LiteralProto
execution_platform (	"1
	HloInputs$
	arguments (2.xla.LiteralProto"e
HloUnoptimizedSnapshot'

hlo_module (2.xla.HloModuleProto"

partitions (2.xla.HloInputs"
HloModuleMetadataProto
canonical_module_id (
module_group_name (	
original_module_id (
partitioned_module_ids (+
pass_metadata (2.xla.HloPassMetadata"
HloPassMetadata
pass_id (
	pass_name (	
pipeline_name (	
dump_filenames (	
module_changed (
	module_id (
module_group_module_ids (
start_timestamp_usec (
end_timestamp_usec	 (-
custom_metadata
 (2.google.protobuf.Any'

kv_metrics (2.xla.KeyValueMetric*S
CustomCallSchedule
SCHEDULE_NONE 
SCHEDULE_LATEST
SCHEDULE_EARLIEST*
CustomCallApiVersion
API_VERSION_UNSPECIFIED 
API_VERSION_ORIGINAL 
API_VERSION_STATUS_RETURNING(
$API_VERSION_STATUS_RETURNING_UNIFIED
API_VERSION_TYPED_FFI*:
Kind
UNDEFINED_ALIAS 
	MAY_ALIAS

MUST_ALIASBbproto3zxla.service.hlo_pb2FNs   replica_groupss   all_reduce_idis_cross_program_prefetchs   8instruction_namei   in   iq   i%!  i'!  ia!  i   i  iB  i  i  i  i  it  i  i  i  iw  iR  i  iU  i  i  i  iG  iJ  i  i
  if  ih  i  i  i  i  i  iF  iI  i  i!  i  i  i  il  i)  io  i  i  i  i  i  i  im  i  io  i  i  i~  i  i  i  i  i  i  i  i   )E__doc__google.protobuf.internalr   _buildergoogle.protobufr   _descriptorr   _descriptor_poolr   _symbol_databaseDefault_sym_dbr    google_dot_protobuf_dot_any__pb2tensorflow.compiler.xla.servicer    xla_dot_service_dot_metrics__pb2tensorflow.compiler.xlar	   xla_dot_xla__data__pb2AddSerializedFile
DESCRIPTORBuildMessageAndEnumDescriptorsglobalsBuildTopDescriptorsAndMessages_USE_C_DESCRIPTORS_options_serialized_options_HLOINSTRUCTIONPROTOfields_by_name _HLOSCHEDULEPROTO_SEQUENCESENTRY_LOGICALBUFFERPROTO_LOCATION_CUSTOMCALLSCHEDULE_serialized_start_serialized_end_CUSTOMCALLAPIVERSION_KIND$_HLOINSTRUCTIONPROTO_SLICEDIMENSIONS_HLOCOMPUTATIONPROTO_HLOSCHEDULEPROTO%_HLOSCHEDULEPROTO_INSTRUCTIONSEQUENCE_HLOINPUTOUTPUTALIASPROTO)_HLOINPUTOUTPUTALIASPROTO_ALIASENTRYPROTO_HLOBUFFERDONORPROTO*_HLOBUFFERDONORPROTO_BUFFERDONORENTRYPROTO_CROSSPROGRAMPREFETCH_STACKFRAMEINDEXPROTO"_STACKFRAMEINDEXPROTO_FILELOCATION _STACKFRAMEINDEXPROTO_STACKFRAME_HLOMODULEPROTO_HLOMODULEPROTO_PROFILEINFO_HLOMODULEPROTO_PROFILETYPE)_HLOMODULEPROTO_PROFILEGENERATIONSTRATEGY_LOGICALBUFFERPROTO_BUFFERALLOCATIONPROTO_BUFFERALLOCATIONPROTO_ASSIGNED_HEAPSIMULATORTRACE_HEAPSIMULATORTRACE_EVENT_HEAPSIMULATORTRACE_EVENT_KIND_HLOMODULEGROUPPROTO_BUFFERASSIGNMENTPROTO"_BUFFERASSIGNMENTPROTO_BUFFERALIAS	_HLOPROTO_HLOSNAPSHOT
_HLOINPUTS_HLOUNOPTIMIZEDSNAPSHOT_HLOMODULEMETADATAPROTO_HLOPASSMETADATA     W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/compiler/xla/service/hlo_pb2.py<module>rP      s   & 8 5 ? ? #

"
"
$ H [ J &%%'99  ;i_  j_
 ' ' '
GI > ' ' '
4I79 U!!U**#2* CG%%&67@NY%%&67KBF%%o6?MX%%o6JNR%%&ABKYd%%&ABV.2"+9A"6MQ--.@AJXc--.@AU(,'&*%*.)(,'%%),('+&9=&87;&6)-('+&&*%$(#:>'98<'759"437"2.2-,0+>B+=<@+;)-('+&?C,>=A,<*.)(,'*.)(,'7;$659$459"437"2$(/#"&/!04/.2-04/.2->B+=<@+;(,'&*%150/3.+/*)-(48!326!1(,'&*%.2-,0+37 215 0)-('+&+/*)-(7;$659$4") )!%, #,#*!*,0+*.),0+*.)%)$#'"o +rN   