o
    i eKK                     @   s  d Z ddlmZ ddlmZ ddlmZ ddlm	Z
 e
 ZddlmZ e dZeee  eede  ejd	krVd
e_de_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,e+_d-e+_d.e,_d+e,_d/e-_d0e-_d1e._d2e._d3e/_d0e/_d4e0_d5e0_d6e_d5e_d7e1_d8e1_d9e2_d8e2_d:e3_d;e3_d<e4_d;e4_d=e5_d;e5_d>e6_d?e6_d@e7_dAe7_dBe8_dAe8_dCe9_dDe9_dEe:_dFe:_dGe;_dHe;_dIe<_dJe<_dKe=_dLe=_dMe>_dNe>_dOe?_dLe?_dPe@_dQe@_d
S d
S )RzGenerated protocol buffer code.    )builder)
descriptor)descriptor_pool)symbol_database)xla_data_pb2s,   
)tensorflow/compiler/xla/service/hlo.protoxla&tensorflow/compiler/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
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.ReplicaGroup
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_group_idN (
async_execution_threadO (		
kQ (*
statistics_vizR (2.xla.StatisticsViz?
SliceDimensions
start (
limit (
stride (B'
%optional_cross_program_prefetch_indexJ
JJJJJ,-J56J./J)*J*+J@AR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"
DynamicParameterBindingProto:
entries (2).xla.DynamicParameterBindingProto.Binding
Binding
dynamic_param_num (
dynamic_param_index (
target_param_num (
target_param_index (
target_param_dim_num ("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D
dynamic_parameter_binding	 (2!.xla.DynamicParameterBindingProto;
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
ProfileInfo5
profile_type (2.xla.HloModuleProto.ProfileType
relative_speedup (*
profile_source (2.xla.ProfileSource0
compilation_event (2.xla.CompilationEvent"E
ProfileType
INVALID 
FLAG

FUSION

LAYOUT
DOT"
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 (	"
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	 ("
EntryFunctionAttributesG
buffers (26.xla.EntryFunctionAttributes.BufferParameterAttributes
result_xla_shape (	

ShapeIndex
indices (
BufferParameterAttributes
lmhlo_params (
lmhlo_params_present (H
lmhlo_param_shape_index (2'.xla.EntryFunctionAttributes.ShapeIndex
lmhlo_constant_name (	
lmhlo_must_alias (C
lmhlo_output_index (2'.xla.EntryFunctionAttributes.ShapeIndex"q
XlaRuntimeExecutableProto-
hlo_module_proto (2.xla.HloModuleProto
obj_file (
mlir_module (	*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'tensorflow.compiler.xla.service.hlo_pb2FNs   all_reduce_ids   is_cross_program_prefetchs   8instruction_namei  i,  i/  i  i  i   [   if  i
  iL
  ii  iR  iU  i-  i  i  i  i0  i  i  i  i   ik  i  iJ  iM  i  i  ii  ik  i  i  i  i^  i`  i  ix  i  i{  i  i  i  iL  i~  i!  iN  i  i  i  i  i  it  iw  i  i  i  i  i  i  id  i/  iL  iO  if  i  )A__doc__google.protobuf.internalr   _buildergoogle.protobufr   _descriptorr   _descriptor_poolr   _symbol_databaseDefault_sym_dbtensorflow.compiler.xlar   2tensorflow_dot_compiler_dot_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_DYNAMICPARAMETERBINDINGPROTO%_DYNAMICPARAMETERBINDINGPROTO_BINDING_CROSSPROGRAMPREFETCH_STACKFRAMEINDEXPROTO"_STACKFRAMEINDEXPROTO_FILELOCATION _STACKFRAMEINDEXPROTO_STACKFRAME_HLOMODULEPROTO_HLOMODULEPROTO_PROFILEINFO_HLOMODULEPROTO_PROFILETYPE_LOGICALBUFFERPROTO_BUFFERALLOCATIONPROTO_BUFFERALLOCATIONPROTO_ASSIGNED_HEAPSIMULATORTRACE_HEAPSIMULATORTRACE_EVENT_HEAPSIMULATORTRACE_EVENT_KIND_HLOMODULEGROUPPROTO_BUFFERASSIGNMENTPROTO"_BUFFERASSIGNMENTPROTO_BUFFERALIAS	_HLOPROTO_HLOSNAPSHOT_HLOMODULEMETADATAPROTO_HLOPASSMETADATA_ENTRYFUNCTIONATTRIBUTES#_ENTRYFUNCTIONATTRIBUTES_SHAPEINDEX2_ENTRYFUNCTIONATTRIBUTES_BUFFERPARAMETERATTRIBUTES_XLARUNTIMEEXECUTABLEPROTO rG   rG   V/var/www/myenv/lib/python3.10/site-packages/tensorflow/compiler/xla/service/hlo_pb2.py<module>   s   
