137 lines
18 KiB
Python
137 lines
18 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||
|
|
# source: xla/xla_data.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()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12xla/xla_data.proto\x12\x03xla\"\xb7\x01\n\rPaddingConfig\x12=\n\ndimensions\x18\x01 \x03(\x0b\x32).xla.PaddingConfig.PaddingConfigDimension\x1ag\n\x16PaddingConfigDimension\x12\x18\n\x10\x65\x64ge_padding_low\x18\x01 \x01(\x03\x12\x19\n\x11\x65\x64ge_padding_high\x18\x02 \x01(\x03\x12\x18\n\x10interior_padding\x18\x03 \x01(\x03\"\x1f\n\tTileProto\x12\x12\n\ndimensions\x18\x01 \x03(\x03\"\x8c\x04\n\x0bLayoutProto\x12*\n\x0f\x64im_level_types\x18\t \x03(\x0e\x32\x11.xla.DimLevelType\x12\x12\n\ndim_unique\x18\r \x03(\x08\x12\x13\n\x0b\x64im_ordered\x18\x0e \x03(\x08\x12\x16\n\x0eminor_to_major\x18\x01 \x03(\x03\x12\x1d\n\x05tiles\x18\x06 \x03(\x0b\x32\x0e.xla.TileProto\x12*\n\"tail_padding_alignment_in_elements\x18\x10 \x01(\x03\x12\x1c\n\x14\x65lement_size_in_bits\x18\x07 \x01(\x03\x12\x14\n\x0cmemory_space\x18\x08 \x01(\x03\x12\x30\n\x14index_primitive_type\x18\x0b \x01(\x0e\x32\x12.xla.PrimitiveType\x12\x32\n\x16pointer_primitive_type\x18\x0c \x01(\x0e\x32\x12.xla.PrimitiveType\x12\'\n\x0ephysical_shape\x18\n \x01(\x0b\x32\x0f.xla.ShapeProto\x12+\n#dynamic_shape_metadata_prefix_bytes\x18\x0f \x01(\x03J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06R\x11padded_dimensionsR\rpadding_valueR\x06\x66ormatR\x13max_sparse_elements\"\xbd\x01\n\nShapeProto\x12(\n\x0c\x65lement_type\x18\x02 \x01(\x0e\x32\x12.xla.PrimitiveType\x12\x12\n\ndimensions\x18\x03 \x03(\x03\x12%\n\x0ctuple_shapes\x18\x04 \x03(\x0b\x32\x0f.xla.ShapeProto\x12 \n\x06layout\x18\x05 \x01(\x0b\x32\x10.xla.LayoutProto\x12\x1c\n\x14is_dynamic_dimension\x18\x06 \x03(\x08J\x04\x08\x01\x10\x02R\x04rank\"r\n\x11ProgramShapeProto\x12#\n\nparameters\x18\x01 \x03(\x0b\x32\x0f.xla.ShapeProto\x12\x1f\n\x06result\x18\x02 \x01(\x0b\x32\x0f.xla.ShapeProto\x12\x17\n\x0fparameter_names\x18\x03 \x03(\t\"D\n\x10\x43omputationStats\x12\x12\n\nflop_count\x18\x01 \x01(\x01\x12\x1c\n\x14transcendental_count\x18\x02 \x01(\x01\"\xcb\x04\n\nOpMetadata\x12\x0f\n\x07op_type\x18\x01 \x01(\t\x12\x0f\n\x07op_name\x18\x02 \x01(\t\x12\x13\n\x0bsource_file\x18\x03 \x01(\t\x12\x13\n\x0bsource_line\x18\x04 \x01(\x05\x12*\n\x0cprofile_type\x18\x05 \x03(\x0e\x32\x10.xla.ProfileTypeB\x02\x18\x01\x12\x18\n\x10\x63reation_pass_id\x18\x06 \x01(\x03\x12 \n\x18logical_creation_pass_id\x18\x07 \x01(\x03\x12\'\n\x1fsize_of_generated_code_in_bytes\x18\x08 \x01(\x03\x12+\n#size_of_memory_working_set_in_bytes\x18\t \x01(\x03\x12\x31\n\x0cprofile_info\x18\n \x01(\x0b\x32\x1b.xla.OpMetadata.ProfileInfo\x12\x19\n\x11\x64\x65\x64uplicated_name\x18\x0c \x01(\t\x12\x17\n\x0fpreserve_layout\x18\r \x01(\x08\x12\x16\n\x0estack_frame_id\x18\x0f \x01(\x05\x1a\xad\x01\n\x0bProfileInfo\x12&\n\x0cprofile_type\x18\x01 \x03(\x0e\x32\x10.xla.ProfileType\x12\x18\n\x10relative_speedup\x18\x02 \x01(\x01\x12*\n\x0eprofile_source\x18\x03 \x01(\x0e\x32\x12.xla.ProfileSource\x12\x30\n\x11\x63ompilation_event\x18\x04 \x01(\x0e\x32\x15.xla.CompilationEventJ\x04\x08\x0e\x10\x0f\"\xe3\x01\n\x10\x45xecutionProfile\x12\x1d\n\x15\x63ompilation_cache_hit\x18\x01 \x01(\x08\x12\x17\n\x0f\x63ompile_time_ms\x18\x02 \x01(\x03\x12\x1b\n\x13\x63ompute_cycle_count\x18\x03 \x01(\x03\x12\x17\n\x0f\x63ompute_time_ns\x18\x04 \x01(\x03\x12$\n\x1c\x63ompute_and_transfer_time_ns\x18\x05 \x01(\x03\x12 \n\x18\x65xecutable_size_in_bytes\x18\x06 \x01(\x03\x12\x19\n\x11profile_cache_hit\x18\x07 \x01(\x08\"!\n\x0f\x45xecutionHandle\x12\x0e\n\x06handle\x18\x01 \x01(\x03\"\"\n\x10GlobalDataHandle\x12\x0e\n\x06handle\x18\x01 \x01(\x03\"4\n\x0c\x44\x65viceHandle\x12\x0e\n\x06handle\x18\x01 \x01(\x03\x12\x14\n\x0c\x64\x65vice_count\x18\x02 \x01(\x03\"\xb4\x01\n\rChannelHandle\x12\x0e\n\x06handle\x18\x01 \x01(\x03\x12,\n\x04type\x18\x02 \x01(\x0e\x32\x1e.xla.ChannelHandle.ChannelType\"e\n\x0b\x43hannelType\x12\x18\n\x14\x43HANNEL_TYPE_INVALID\x10\x00\x12\x14\n\x10\x44\x45VICE_TO_DEVICE\x10\x01\x12\x12\n\x0e\x44\x45VICE_TO_HOST\x10\x02\x12\x12\n\x0eHOST_TO_DEVICE\x10\x03\"\xc5\x01\n\x15\x44\x65viceAssignmentProto\x12\x15\n\rreplica_count\x18\x01 \x01(\x05\x12\x19\n\x1
|
||
|
|
|
||
|
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
||
|
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'xla.xla_data_pb2', globals())
|
||
|
|
if _descriptor._USE_C_DESCRIPTORS == False:
|
||
|
|
|
||
|
|
DESCRIPTOR._options = None
|
||
|
|
DESCRIPTOR._serialized_options = b'\370\001\001'
|
||
|
|
_OPMETADATA.fields_by_name['profile_type']._options = None
|
||
|
|
_OPMETADATA.fields_by_name['profile_type']._serialized_options = b'\030\001'
|
||
|
|
_FRONTENDATTRIBUTES_MAPENTRY._options = None
|
||
|
|
_FRONTENDATTRIBUTES_MAPENTRY._serialized_options = b'8\001'
|
||
|
|
_PRIMITIVETYPE._serialized_start=5709
|
||
|
|
_PRIMITIVETYPE._serialized_end=6020
|
||
|
|
_DIMLEVELTYPE._serialized_start=6022
|
||
|
|
_DIMLEVELTYPE._serialized_end=6116
|
||
|
|
_PROFILETYPE._serialized_start=6118
|
||
|
|
_PROFILETYPE._serialized_end=6179
|
||
|
|
_PROFILESOURCE._serialized_start=6181
|
||
|
|
_PROFILESOURCE._serialized_end=6287
|
||
|
|
_COMPILATIONEVENT._serialized_start=6290
|
||
|
|
_COMPILATIONEVENT._serialized_end=6423
|
||
|
|
_PADDINGTYPE._serialized_start=6425
|
||
|
|
_PADDINGTYPE._serialized_end=6496
|
||
|
|
_FFTTYPE._serialized_start=6498
|
||
|
|
_FFTTYPE._serialized_end=6547
|
||
|
|
_SPARSITYTYPE._serialized_start=6549
|
||
|
|
_SPARSITYTYPE._serialized_end=6614
|
||
|
|
_RANDOMDISTRIBUTION._serialized_start=6616
|
||
|
|
_RANDOMDISTRIBUTION._serialized_end=6686
|
||
|
|
_RANDOMALGORITHM._serialized_start=6688
|
||
|
|
_RANDOMALGORITHM._serialized_end=6757
|
||
|
|
_PADDINGCONFIG._serialized_start=28
|
||
|
|
_PADDINGCONFIG._serialized_end=211
|
||
|
|
_PADDINGCONFIG_PADDINGCONFIGDIMENSION._serialized_start=108
|
||
|
|
_PADDINGCONFIG_PADDINGCONFIGDIMENSION._serialized_end=211
|
||
|
|
_TILEPROTO._serialized_start=213
|
||
|
|
_TILEPROTO._serialized_end=244
|
||
|
|
_LAYOUTPROTO._serialized_start=247
|
||
|
|
_LAYOUTPROTO._serialized_end=771
|
||
|
|
_SHAPEPROTO._serialized_start=774
|
||
|
|
_SHAPEPROTO._serialized_end=963
|
||
|
|
_PROGRAMSHAPEPROTO._serialized_start=965
|
||
|
|
_PROGRAMSHAPEPROTO._serialized_end=1079
|
||
|
|
_COMPUTATIONSTATS._serialized_start=1081
|
||
|
|
_COMPUTATIONSTATS._serialized_end=1149
|
||
|
|
_OPMETADATA._serialized_start=1152
|
||
|
|
_OPMETADATA._serialized_end=1739
|
||
|
|
_OPMETADATA_PROFILEINFO._serialized_start=1560
|
||
|
|
_OPMETADATA_PROFILEINFO._serialized_end=1733
|
||
|
|
_EXECUTIONPROFILE._serialized_start=1742
|
||
|
|
_EXECUTIONPROFILE._serialized_end=1969
|
||
|
|
_EXECUTIONHANDLE._serialized_start=1971
|
||
|
|
_EXECUTIONHANDLE._serialized_end=2004
|
||
|
|
_GLOBALDATAHANDLE._serialized_start=2006
|
||
|
|
_GLOBALDATAHANDLE._serialized_end=2040
|
||
|
|
_DEVICEHANDLE._serialized_start=2042
|
||
|
|
_DEVICEHANDLE._serialized_end=2094
|
||
|
|
_CHANNELHANDLE._serialized_start=2097
|
||
|
|
_CHANNELHANDLE._serialized_end=2277
|
||
|
|
_CHANNELHANDLE_CHANNELTYPE._serialized_start=2176
|
||
|
|
_CHANNELHANDLE_CHANNELTYPE._serialized_end=2277
|
||
|
|
_DEVICEASSIGNMENTPROTO._serialized_start=2280
|
||
|
|
_DEVICEASSIGNMENTPROTO._serialized_end=2477
|
||
|
|
_DEVICEASSIGNMENTPROTO_COMPUTATIONDEVICE._serialized_start=2430
|
||
|
|
_DEVICEASSIGNMENTPROTO_COMPUTATIONDEVICE._serialized_end=2477
|
||
|
|
_LITERALPROTO._serialized_start=2480
|
||
|
|
_LITERALPROTO._serialized_end=2932
|
||
|
|
_WINDOWDIMENSION._serialized_start=2935
|
||
|
|
_WINDOWDIMENSION._serialized_end=3098
|
||
|
|
_WINDOW._serialized_start=3100
|
||
|
|
_WINDOW._serialized_end=3150
|
||
|
|
_GATHERDIMENSIONNUMBERS._serialized_start=3152
|
||
|
|
_GATHERDIMENSIONNUMBERS._serialized_end=3278
|
||
|
|
_SCATTERDIMENSIONNUMBERS._serialized_start=3281
|
||
|
|
_SCATTERDIMENSIONNUMBERS._serialized_end=3428
|
||
|
|
_CONVOLUTIONDIMENSIONNUMBERS._serialized_start=3431
|
||
|
|
_CONVOLUTIONDIMENSIONNUMBERS._serialized_end=3775
|
||
|
|
_DOTDIMENSIONNUMBERS._serialized_start=3778
|
||
|
|
_DOTDIMENSIONNUMBERS._serialized_end=3931
|
||
|
|
_SPARSITYDESCRIPTOR._serialized_start=3933
|
||
|
|
_SPARSITYDESCRIPTOR._serialized_end=4042
|
||
|
|
_TRIANGULARSOLVEOPTIONS._serialized_start=4045
|
||
|
|
_TRIANGULARSOLVEOPTIONS._serialized_end=4268
|
||
|
|
_TRIANGULARSOLVEOPTIONS_TRANSPOSE._serialized_start=4188
|
||
|
|
_TRIANGULARSOLVEOPTIONS_TRANSPOSE._serialized_end=4268
|
||
|
|
_CHOLESKYOPTIONS._serialized_start=4270
|
||
|
|
_CHOLESKYOPTIONS._serialized_end=4302
|
||
|
|
_SORTOPTIONS._serialized_start=4304
|
||
|
|
_SORTOPTIONS._serialized_end=4337
|
||
|
|
_FRONTENDATTRIBUTES._serialized_start=4339
|
||
|
|
_FRONTENDATTRIBUTES._serialized_end=4450
|
||
|
|
_FRONTENDATTRIBUTES_MAPENTRY._serialized_start=4408
|
||
|
|
_FRONTENDATTRIBUTES_MAPENTRY._serialized_end=4450
|
||
|
|
_STATISTIC._serialized_start=4452
|
||
|
|
_STATISTIC._serialized_end=4500
|
||
|
|
_STATISTICSVIZ._serialized_start=4502
|
||
|
|
_STATISTICSVIZ._serialized_end=4586
|
||
|
|
_OPSHARDING._serialized_start=4589
|
||
|
|
_OPSHARDING._serialized_end=5184
|
||
|
|
_OPSHARDING_TYPE._serialized_start=5066
|
||
|
|
_OPSHARDING_TYPE._serialized_end=5148
|
||
|
|
_OPSHARDING_SHARDGROUPTYPE._serialized_start=5150
|
||
|
|
_OPSHARDING_SHARDGROUPTYPE._serialized_end=5184
|
||
|
|
_REPLICAGROUP._serialized_start=5186
|
||
|
|
_REPLICAGROUP._serialized_end=5221
|
||
|
|
_SOURCETARGET._serialized_start=5223
|
||
|
|
_SOURCETARGET._serialized_end=5269
|
||
|
|
_PRECISIONCONFIG._serialized_start=5272
|
||
|
|
_PRECISIONCONFIG._serialized_end=5416
|
||
|
|
_PRECISIONCONFIG_PRECISION._serialized_start=5350
|
||
|
|
_PRECISIONCONFIG_PRECISION._serialized_end=5416
|
||
|
|
_PARAMETERREPLICATION._serialized_start=5418
|
||
|
|
_PARAMETERREPLICATION._serialized_end=5476
|
||
|
|
_WHILELOOPBACKENDCONFIG._serialized_start=5478
|
||
|
|
_WHILELOOPBACKENDCONFIG._serialized_end=5601
|
||
|
|
_WHILELOOPBACKENDCONFIG_KNOWNTRIPCOUNT._serialized_start=5574
|
||
|
|
_WHILELOOPBACKENDCONFIG_KNOWNTRIPCOUNT._serialized_end=5601
|
||
|
|
_OUTPUTOPERANDALIASING._serialized_start=5603
|
||
|
|
_OUTPUTOPERANDALIASING._serialized_end=5706
|
||
|
|
# @@protoc_insertion_point(module_scope)
|