105 lines
18 KiB
Python
105 lines
18 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||
|
|
# source: xla/service/hlo.proto
|
||
|
|
"""Generated protocol buffer code."""
|
||
|
|
from google.protobuf.internal import builder as _builder
|
||
|
|
from google.protobuf import descriptor as _descriptor
|
||
|
|
from google.protobuf import descriptor_pool as _descriptor_pool
|
||
|
|
from google.protobuf import symbol_database as _symbol_database
|
||
|
|
# @@protoc_insertion_point(imports)
|
||
|
|
|
||
|
|
_sym_db = _symbol_database.Default()
|
||
|
|
|
||
|
|
|
||
|
|
from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
|
||
|
|
from tensorflow.compiler.xla import xla_data_pb2 as xla_dot_xla__data__pb2
|
||
|
|
|
||
|
|
|
||
|
|
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15xla/service/hlo.proto\x12\x03xla\x1a\x19google/protobuf/any.proto\x1a\x12xla/xla_data.proto\"\xc5\x16\n\x13HloInstructionProto\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06opcode\x18\x02 \x01(\t\x12\x1e\n\x05shape\x18\x03 \x01(\x0b\x32\x0f.xla.ShapeProto\x12!\n\x08metadata\x18\x07 \x01(\x0b\x32\x0f.xla.OpMetadata\x12\"\n\x07literal\x18\x08 \x01(\x0b\x32\x11.xla.LiteralProto\x12\x18\n\x10parameter_number\x18\t \x01(\x03\x12\x13\n\x0b\x66usion_kind\x18\x0b \x01(\t\x12\x13\n\x0btuple_index\x18\r \x01(\x03\x12\x12\n\ndimensions\x18\x0e \x03(\x03\x12\x1b\n\x06window\x18\x0f \x01(\x0b\x32\x0b.xla.Window\x12G\n\x1d\x63onvolution_dimension_numbers\x18\x10 \x01(\x0b\x32 .xla.ConvolutionDimensionNumbers\x12\x1b\n\x13\x66\x65\x61ture_group_count\x18\x32 \x01(\x03\x12\x19\n\x11\x62\x61tch_group_count\x18: \x01(\x03\x12\x42\n\x10slice_dimensions\x18\x11 \x03(\x0b\x32(.xla.HloInstructionProto.SliceDimensions\x12\x15\n\rexponent_bits\x18\x12 \x01(\x05\x12\x15\n\rmantissa_bits\x18\x13 \x01(\x05\x12\x1b\n\x13\x64ynamic_slice_sizes\x18\x14 \x03(\x03\x12*\n\x0epadding_config\x18\x15 \x01(\x0b\x32\x12.xla.PaddingConfig\x12\x16\n\x0eoutfeed_config\x18\x16 \x01(\x0c\x12-\n\x0c\x64istribution\x18\x17 \x01(\x0e\x32\x17.xla.RandomDistribution\x12\x0f\n\x07\x65psilon\x18\x18 \x01(\x02\x12\x15\n\rfeature_index\x18\x19 \x01(\x03\x12\x12\n\nchannel_id\x18\x1a \x01(\x03\x12\x15\n\rinfeed_config\x18\x1b \x01(\x0c\x12\x1a\n\x12\x63ustom_call_target\x18\x1c \x01(\t\x12&\n\routfeed_shape\x18\x1d \x01(\x0b\x32\x0f.xla.ShapeProto\x12\x37\n\x15\x64ot_dimension_numbers\x18\x1e \x01(\x0b\x32\x18.xla.DotDimensionNumbers\x12\x1e\n\x08\x66\x66t_type\x18\x1f \x01(\x0e\x32\x0c.xla.FftType\x12\x12\n\nfft_length\x18 \x03(\x03\x12\x1c\n\x14\x63omparison_direction\x18? \x01(\t\x12=\n\x18gather_dimension_numbers\x18! \x01(\x0b\x32\x1b.xla.GatherDimensionNumbers\x12\x1a\n\x12gather_slice_sizes\x18\" \x03(\x03\x12\n\n\x02id\x18# \x01(\x03\x12\x13\n\x0boperand_ids\x18$ \x03(\x03\x12\x1f\n\x17\x63ontrol_predecessor_ids\x18% \x03(\x03\x12\x1e\n\x16\x63\x61lled_computation_ids\x18& \x03(\x03\x12!\n\x08sharding\x18( \x01(\x0b\x32\x0f.xla.OpSharding\x12\x16\n\x0e\x62\x61\x63kend_config\x18+ \x01(\x0c\x12)\n\x0ereplica_groups\x18\x31 \x03(\x0b\x32\x11.xla.ReplicaGroup\x12\x19\n\rall_reduce_id\x18- \x01(\x03\x42\x02\x18\x01\x12\x1d\n\x15use_global_device_ids\x18G \x01(\x08\x12\x18\n\x10is_host_transfer\x18/ \x01(\x08\x12\x11\n\tis_stable\x18< \x01(\x08\x12?\n\x19scatter_dimension_numbers\x18\x30 \x01(\x0b\x32\x1c.xla.ScatterDimensionNumbers\x12.\n\x10precision_config\x18\x33 \x01(\x0b\x32\x14.xla.PrecisionConfig\x12.\n\x13source_target_pairs\x18\x34 \x03(\x0b\x32\x11.xla.SourceTarget\x12.\n\x15\x64omain_entry_sharding\x18\x36 \x01(\x0b\x32\x0f.xla.OpSharding\x12-\n\x14\x64omain_exit_sharding\x18\x37 \x01(\x0b\x32\x0f.xla.OpSharding\x12\x18\n\x10\x63onstrain_layout\x18\x38 \x01(\x08\x12\x33\n\x1aoperand_shapes_with_layout\x18\x39 \x03(\x0b\x32\x0f.xla.ShapeProto\x12=\n\x18triangular_solve_options\x18; \x01(\x0b\x32\x1b.xla.TriangularSolveOptions\x12.\n\x10\x63holesky_options\x18> \x01(\x0b\x32\x14.xla.CholeskyOptions\x12\x38\n\x15parameter_replication\x18= \x01(\x0b\x32\x19.xla.ParameterReplication\x12#\n\x1b\x63ustom_call_has_side_effect\x18\x41 \x01(\x08\x12;\n\x17output_operand_aliasing\x18J \x03(\x0b\x32\x1a.xla.OutputOperandAliasing\x12\x35\n\x14\x63ustom_call_schedule\x18L \x01(\x0e\x32\x17.xla.CustomCallSchedule\x12\r\n\x05\x64\x65lta\x18\x42 \x01(\x03\x12\x1a\n\x12indices_are_sorted\x18\x43 \x01(\x08\x12\x34\n\x13\x66rontend_attributes\x18\x44 \x01(\x0b\x32\x17.xla.FrontendAttributes\x12\x16\n\x0eunique_indices\x18\x45 \x01(\x08\x12+\n\rrng_algorithm\x18\x46 \x01(\x0e\x32\x14.xla.RandomAlgorithm\x12\x17\n\x0f\x63omparison_type\x18H \x01(\t\x12%\n\x19is_cross_program_prefetch\x18I \x01(\x08\x42\x02\x18\x01\x12&\n\x1c\x63ross_program_prefetch_index\x18P \x01(\x05H\x00\x12&\n\x0cpadding_type\x18K \x01(\x0e\x32\x10.xla.PaddingType\x12:\n\x17\x63ustom_call_api_version\x18M \x01(\x0e\x32\x19.xla.CustomCallApi
|
||
|
|
|
||
|
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
||
|
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'xla.service.hlo_pb2', globals())
|
||
|
|
if _descriptor._USE_C_DESCRIPTORS == False:
|
||
|
|
|
||
|
|
DESCRIPTOR._options = None
|
||
|
|
DESCRIPTOR._serialized_options = b'\370\001\001'
|
||
|
|
_HLOINSTRUCTIONPROTO.fields_by_name['all_reduce_id']._options = None
|
||
|
|
_HLOINSTRUCTIONPROTO.fields_by_name['all_reduce_id']._serialized_options = b'\030\001'
|
||
|
|
_HLOINSTRUCTIONPROTO.fields_by_name['is_cross_program_prefetch']._options = None
|
||
|
|
_HLOINSTRUCTIONPROTO.fields_by_name['is_cross_program_prefetch']._serialized_options = b'\030\001'
|
||
|
|
_HLOSCHEDULEPROTO_SEQUENCESENTRY._options = None
|
||
|
|
_HLOSCHEDULEPROTO_SEQUENCESENTRY._serialized_options = b'8\001'
|
||
|
|
_LOGICALBUFFERPROTO_LOCATION.fields_by_name['instruction_name']._options = None
|
||
|
|
_LOGICALBUFFERPROTO_LOCATION.fields_by_name['instruction_name']._serialized_options = b'\030\001'
|
||
|
|
_CUSTOMCALLSCHEDULE._serialized_start=7563
|
||
|
|
_CUSTOMCALLSCHEDULE._serialized_end=7646
|
||
|
|
_CUSTOMCALLAPIVERSION._serialized_start=7649
|
||
|
|
_CUSTOMCALLAPIVERSION._serialized_end=7829
|
||
|
|
_KIND._serialized_start=7831
|
||
|
|
_KIND._serialized_end=7889
|
||
|
|
_HLOINSTRUCTIONPROTO._serialized_start=78
|
||
|
|
_HLOINSTRUCTIONPROTO._serialized_end=2963
|
||
|
|
_HLOINSTRUCTIONPROTO_SLICEDIMENSIONS._serialized_start=2600
|
||
|
|
_HLOINSTRUCTIONPROTO_SLICEDIMENSIONS._serialized_end=2663
|
||
|
|
_HLOCOMPUTATIONPROTO._serialized_start=2966
|
||
|
|
_HLOCOMPUTATIONPROTO._serialized_end=3199
|
||
|
|
_HLOSCHEDULEPROTO._serialized_start=3202
|
||
|
|
_HLOSCHEDULEPROTO._serialized_end=3418
|
||
|
|
_HLOSCHEDULEPROTO_INSTRUCTIONSEQUENCE._serialized_start=3279
|
||
|
|
_HLOSCHEDULEPROTO_INSTRUCTIONSEQUENCE._serialized_end=3325
|
||
|
|
_HLOSCHEDULEPROTO_SEQUENCESENTRY._serialized_start=3327
|
||
|
|
_HLOSCHEDULEPROTO_SEQUENCESENTRY._serialized_end=3418
|
||
|
|
_HLOINPUTOUTPUTALIASPROTO._serialized_start=3421
|
||
|
|
_HLOINPUTOUTPUTALIASPROTO._serialized_end=3640
|
||
|
|
_HLOINPUTOUTPUTALIASPROTO_ALIASENTRYPROTO._serialized_start=3513
|
||
|
|
_HLOINPUTOUTPUTALIASPROTO_ALIASENTRYPROTO._serialized_end=3640
|
||
|
|
_HLOBUFFERDONORPROTO._serialized_start=3643
|
||
|
|
_HLOBUFFERDONORPROTO._serialized_end=3811
|
||
|
|
_HLOBUFFERDONORPROTO_BUFFERDONORENTRYPROTO._serialized_start=3731
|
||
|
|
_HLOBUFFERDONORPROTO_BUFFERDONORENTRYPROTO._serialized_end=3811
|
||
|
|
_CROSSPROGRAMPREFETCH._serialized_start=3813
|
||
|
|
_CROSSPROGRAMPREFETCH._serialized_end=3885
|
||
|
|
_STACKFRAMEINDEXPROTO._serialized_start=3888
|
||
|
|
_STACKFRAMEINDEXPROTO._serialized_end=4237
|
||
|
|
_STACKFRAMEINDEXPROTO_FILELOCATION._serialized_start=4080
|
||
|
|
_STACKFRAMEINDEXPROTO_FILELOCATION._serialized_end=4172
|
||
|
|
_STACKFRAMEINDEXPROTO_STACKFRAME._serialized_start=4174
|
||
|
|
_STACKFRAMEINDEXPROTO_STACKFRAME._serialized_end=4237
|
||
|
|
_HLOMODULEPROTO._serialized_start=4240
|
||
|
|
_HLOMODULEPROTO._serialized_end=5342
|
||
|
|
_HLOMODULEPROTO_PROFILEINFO._serialized_start=5029
|
||
|
|
_HLOMODULEPROTO_PROFILEINFO._serialized_end=5238
|
||
|
|
_HLOMODULEPROTO_PROFILETYPE._serialized_start=5240
|
||
|
|
_HLOMODULEPROTO_PROFILETYPE._serialized_end=5309
|
||
|
|
_LOGICALBUFFERPROTO._serialized_start=5345
|
||
|
|
_LOGICALBUFFERPROTO._serialized_end=5553
|
||
|
|
_LOGICALBUFFERPROTO_LOCATION._serialized_start=5462
|
||
|
|
_LOGICALBUFFERPROTO_LOCATION._serialized_end=5553
|
||
|
|
_BUFFERALLOCATIONPROTO._serialized_start=5556
|
||
|
|
_BUFFERALLOCATIONPROTO._serialized_end=5932
|
||
|
|
_BUFFERALLOCATIONPROTO_ASSIGNED._serialized_start=5865
|
||
|
|
_BUFFERALLOCATIONPROTO_ASSIGNED._serialized_end=5932
|
||
|
|
_HEAPSIMULATORTRACE._serialized_start=5935
|
||
|
|
_HEAPSIMULATORTRACE._serialized_end=6277
|
||
|
|
_HEAPSIMULATORTRACE_EVENT._serialized_start=6071
|
||
|
|
_HEAPSIMULATORTRACE_EVENT._serialized_end=6277
|
||
|
|
_HEAPSIMULATORTRACE_EVENT_KIND._serialized_start=6234
|
||
|
|
_HEAPSIMULATORTRACE_EVENT_KIND._serialized_end=6277
|
||
|
|
_HLOMODULEGROUPPROTO._serialized_start=6279
|
||
|
|
_HLOMODULEGROUPPROTO._serialized_end=6356
|
||
|
|
_BUFFERASSIGNMENTPROTO._serialized_start=6359
|
||
|
|
_BUFFERASSIGNMENTPROTO._serialized_end=6701
|
||
|
|
_BUFFERASSIGNMENTPROTO_BUFFERALIAS._serialized_start=6610
|
||
|
|
_BUFFERASSIGNMENTPROTO_BUFFERALIAS._serialized_end=6701
|
||
|
|
_HLOPROTO._serialized_start=6703
|
||
|
|
_HLOPROTO._serialized_end=6829
|
||
|
|
_HLOSNAPSHOT._serialized_start=6832
|
||
|
|
_HLOSNAPSHOT._serialized_end=6974
|
||
|
|
_HLOMODULEMETADATAPROTO._serialized_start=6977
|
||
|
|
_HLOMODULEMETADATAPROTO._serialized_end=7162
|
||
|
|
_HLOPASSMETADATA._serialized_start=7165
|
||
|
|
_HLOPASSMETADATA._serialized_end=7446
|
||
|
|
_XLARUNTIMEEXECUTABLEPROTO._serialized_start=7448
|
||
|
|
_XLARUNTIMEEXECUTABLEPROTO._serialized_end=7561
|
||
|
|
# @@protoc_insertion_point(module_scope)
|