# -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # NO CHECKED-IN PROTOBUF GENCODE # source: xla/xla_data.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, '', 'xla/xla_data.proto' ) # @@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\"<\n\x10SplitConfigProto\x12\x11\n\tdimension\x18\x01 \x01(\x03\x12\x15\n\rsplit_indices\x18\x02 \x03(\x03\"\xba\x04\n\x0bLayoutProto\x12\x16\n\x0eminor_to_major\x18\x01 \x03(\x03\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\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(\x03\x12,\n\rsplit_configs\x18\x11 \x03(\x0b\x32\x15.xla.SplitConfigProtoJ\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\x1c\n\x14is_dynamic_dimension\x18\x06 \x03(\x08\x12%\n\x0ctuple_shapes\x18\x04 \x03(\x0b\x32\x0f.xla.ShapeProto\x12 \n\x06layout\x18\x05 \x01(\x0b\x32\x10.xla.LayoutProtoJ\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\"\xe3\x05\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\x17\n\x0fsource_end_line\x18\x11 \x01(\x05\x12\x15\n\rsource_column\x18\x12 \x01(\x05\x12\x19\n\x11source_end_column\x18\x13 \x01(\x05\x12*\n\x0cprofile_type\x18\x05 \x03(\x0e\x32\x10.xla.ProfileTypeB\x02\x18\x01\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\x16\n\x0estack_frame_id\x18\x0f \x01(\x05\x12\x17\n\x0fscheduling_name\x18\x10 \x01(\t\x1a\xf2\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.CompilationEvent\x12\x43\n\x1bprofile_generation_strategy\x18\x05 \x01(\x0e\x32\x1e.xla.ProfileGenerationStrategyJ\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x0b\x10\x0cJ\x04\x08\r\x10\x0eJ\x04\x08\x0e\x10\x0fR\x10\x63reation_pass_idR\x18logical_creation_pass_id\"\x80\x02\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\x12\x1b\n\x13warmup_run_executed\x18\x08 \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\x11\x63omputation_count\x18\x02 \x01(\x05\x12I\n\x13\x63omputation_devices\x18\x03 \x03(\x0b\x32,.xla.DeviceAssignmentProto.ComputationDevice\x1a/\n\x11\x43omputationDevice\x12\x1a\n\x12replica_device_ids\x18\x01 \x03(\x03\"\xc1\x04\n\x0cLiteralProto\x12\x1e\n\x05shape\x18\x01 \x01(\x0b\x32\x0f.xla.ShapeProto\x12\r\n\x05preds\x18\x02 \x03(\x08\x12\x0b\n\x03s1s\x18\x1e \x01(\x0c\x12\x0b\n\x03s2s\x18\x1a \x01(\x0c\x12\x0b\n\x03s4s\x18\x15 \x01(\x0c\x12\x0b\n\x03s8s\x18\x0f \x01(\x0c\x12\x0b\n\x03u1s\x18\x1f \x01(\x0c\x12\x0b\n\x03u2s\x18\x1b \x01(\x0c\x12\x0b\n\x03u4s\x18\x16 \x01(\x0c\x12\x0b\n\x03u8s\x18\x03 \x01(\x0c\x12\x0c\n\x04s32s\x18\x04 \x03(\x05\x12\x0c\n\x04s64s\x18\x05 \x03(\x03\x12\x0c\n\x04u32s\x18\x06 \x03(\r\x12\x0c\n\x04u64s\x18\x07 \x03(\x04\x12\x0c\n\x04\x66\x33\x32s\x18\x08 \x03(\x02\x12\x0c\n\x04\x66\x36\x34s\x18\t \x03(\x01\x12\x0c\n\x04\x63\x36\x34s\x18\x0c \x03(\x02\x12\r\n\x05\x63\x31\x32\x38s\x18\x12 \x03(\x01\x12)\n\x0etuple_literals\x18\n \x03(\x0b\x32\x11.xla.LiteralProto\x12\x0c\n\x04\x66\x31\x36s\x18\x0b \x01(\x0c\x12\r\n\x05\x62\x66\x31\x36s\x18\r \x01(\x0c\x12\x0c\n\x04u16s\x18\x10 \x01(\x0c\x12\x0c\n\x04s16s\x18\x11 \x01(\x0c\x12\x11\n\tf4e2m1fns\x18 \x01(\x0c\x12\x0f\n\x07\x66\x38\x65\x33m4s\x18\x1d \x01(\x0c\x12\x16\n\x0e\x66\x38\x65\x34m3b11fnuzs\x18\x17 \x01(\x0c\x12\x11\n\tf8e4m3fns\x18\x14 \x01(\x0c\x12\x13\n\x0b\x66\x38\x65\x34m3fnuzs\x18\x19 \x01(\x0c\x12\x0f\n\x07\x66\x38\x65\x34m3s\x18\x1c \x01(\x0c\x12\x13\n\x0b\x66\x38\x65\x35m2fnuzs\x18\x18 \x01(\x0c\x12\x0f\n\x07\x66\x38\x65\x35m2s\x18\x13 \x01(\x0c\x12\x12\n\nf8e8m0fnus\x18! \x01(\x0c\x12\x16\n\x0esparse_indices\x18\x0e \x03(\x03\"\xa3\x01\n\x0fWindowDimension\x12\x0c\n\x04size\x18\x01 \x01(\x03\x12\x0e\n\x06stride\x18\x02 \x01(\x03\x12\x13\n\x0bpadding_low\x18\x03 \x01(\x03\x12\x14\n\x0cpadding_high\x18\x04 \x01(\x03\x12\x17\n\x0fwindow_dilation\x18\x05 \x01(\x03\x12\x15\n\rbase_dilation\x18\x06 \x01(\x03\x12\x17\n\x0fwindow_reversal\x18\x07 \x01(\x08\"2\n\x06Window\x12(\n\ndimensions\x18\x01 \x03(\x0b\x32\x14.xla.WindowDimension\"\xc2\x01\n\x16GatherDimensionNumbers\x12\x13\n\x0boffset_dims\x18\x01 \x03(\x03\x12\x1c\n\x14\x63ollapsed_slice_dims\x18\x02 \x03(\x03\x12\x17\n\x0fstart_index_map\x18\x03 \x03(\x03\x12\x18\n\x10index_vector_dim\x18\x04 \x01(\x03\x12\x1d\n\x15operand_batching_dims\x18\x05 \x03(\x03\x12#\n\x1bstart_indices_batching_dims\x18\x06 \x03(\x03\"\xd7\x01\n\x17ScatterDimensionNumbers\x12\x1a\n\x12update_window_dims\x18\x01 \x03(\x03\x12\x1c\n\x14inserted_window_dims\x18\x02 \x03(\x03\x12$\n\x1cscatter_dims_to_operand_dims\x18\x03 \x03(\x03\x12\x18\n\x10index_vector_dim\x18\x04 \x01(\x03\x12\x1b\n\x13input_batching_dims\x18\x05 \x03(\x03\x12%\n\x1dscatter_indices_batching_dims\x18\x06 \x03(\x03\"\xd8\x02\n\x1b\x43onvolutionDimensionNumbers\x12\x1d\n\x15input_batch_dimension\x18\x07 \x01(\x03\x12\x1f\n\x17input_feature_dimension\x18\x08 \x01(\x03\x12 \n\x18input_spatial_dimensions\x18\x0b \x03(\x03\x12&\n\x1ekernel_input_feature_dimension\x18\x03 \x01(\x03\x12\'\n\x1fkernel_output_feature_dimension\x18\x04 \x01(\x03\x12!\n\x19kernel_spatial_dimensions\x18\x06 \x03(\x03\x12\x1e\n\x16output_batch_dimension\x18\t \x01(\x03\x12 \n\x18output_feature_dimension\x18\n \x01(\x03\x12!\n\x19output_spatial_dimensions\x18\x0c \x03(\x03\"\x99\x01\n\x13\x44otDimensionNumbers\x12\"\n\x1alhs_contracting_dimensions\x18\x01 \x03(\x03\x12\"\n\x1arhs_contracting_dimensions\x18\x02 \x03(\x03\x12\x1c\n\x14lhs_batch_dimensions\x18\x03 \x03(\x03\x12\x1c\n\x14rhs_batch_dimensions\x18\x04 \x03(\x03\"\x91\x01\n\x19RaggedDotDimensionNumbers\x12\x37\n\x15\x64ot_dimension_numbers\x18\x01 \x01(\x0b\x32\x18.xla.DotDimensionNumbers\x12\x1d\n\x15lhs_ragged_dimensions\x18\x02 \x03(\x03\x12\x1c\n\x14rhs_group_dimensions\x18\x03 \x03(\x03\"\xdf\x01\n\x16TriangularSolveOptions\x12\x11\n\tleft_side\x18\x01 \x01(\x08\x12\r\n\x05lower\x18\x02 \x01(\x08\x12\x15\n\runit_diagonal\x18\x03 \x01(\x08\x12:\n\x0btranspose_a\x18\x04 \x01(\x0e\x32%.xla.TriangularSolveOptions.Transpose\"P\n\tTranspose\x12\x15\n\x11TRANSPOSE_INVALID\x10\x00\x12\x10\n\x0cNO_TRANSPOSE\x10\x01\x12\r\n\tTRANSPOSE\x10\x02\x12\x0b\n\x07\x41\x44JOINT\x10\x03\" \n\x0f\x43holeskyOptions\x12\r\n\x05lower\x18\x01 \x01(\x08\"!\n\x0bSortOptions\x12\x12\n\ndescending\x18\x01 \x01(\x08\"o\n\x12\x46rontendAttributes\x12-\n\x03map\x18\x01 \x03(\x0b\x32 .xla.FrontendAttributes.MapEntry\x1a*\n\x08MapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"0\n\tStatistic\x12\x11\n\tstat_name\x18\x01 \x01(\t\x12\x10\n\x08stat_val\x18\x02 \x01(\x01\"T\n\rStatisticsViz\x12\x1f\n\x17stat_index_to_visualize\x18\x01 \x01(\x03\x12\"\n\nstatistics\x18\x02 \x03(\x0b\x32\x0e.xla.Statistic\"n\n\tMeshProto\x12%\n\x04\x61xes\x18\x01 \x03(\x0b\x32\x17.xla.MeshProto.MeshAxis\x12\x12\n\ndevice_ids\x18\x02 \x03(\x03\x1a&\n\x08MeshAxis\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04size\x18\x02 \x01(\x03\"\x9b\x01\n\x0c\x41xisRefProto\x12\x17\n\x0fmesh_axis_index\x18\x01 \x01(\x03\x12\x35\n\rsub_axis_info\x18\x02 \x01(\x0b\x32\x19.xla.AxisRefProto.SubAxisH\x00\x88\x01\x01\x1a)\n\x07SubAxis\x12\x10\n\x08pre_size\x18\x01 \x01(\x03\x12\x0c\n\x04size\x18\x02 \x01(\x03\x42\x10\n\x0e_sub_axis_info\"\xe5\x02\n\x12NamedShardingProto\x12\x1c\n\x04mesh\x18\x02 \x01(\x0b\x32\x0e.xla.MeshProto\x12@\n\rdim_shardings\x18\x03 \x03(\x0b\x32).xla.NamedShardingProto.DimensionSharding\x12*\n\x0freplicated_axes\x18\x04 \x03(\x0b\x32\x11.xla.AxisRefProto\x12)\n\x0eunreduced_axes\x18\x05 \x03(\x0b\x32\x11.xla.AxisRefProto\x12&\n\x0bmanual_axes\x18\x07 \x03(\x0b\x32\x11.xla.AxisRefProto\x12!\n\x08metadata\x18\x06 \x03(\x0b\x32\x0f.xla.OpMetadata\x1aG\n\x11\x44imensionSharding\x12\x1f\n\x04\x61xes\x18\x01 \x03(\x0b\x32\x11.xla.AxisRefProto\x12\x11\n\tis_closed\x18\x02 \x01(\x08J\x04\x08\x01\x10\x02\"\xab\x05\n\nOpSharding\x12\"\n\x04type\x18\x01 \x01(\x0e\x32\x14.xla.OpSharding.Type\x12#\n\ntile_shape\x18\x02 \x01(\x0b\x32\x0f.xla.ShapeProto\x12\"\n\x1atile_assignment_dimensions\x18\x03 \x03(\x03\x12\x1f\n\x17tile_assignment_devices\x18\x04 \x03(\x03\x12(\n\x0ftuple_shardings\x18\x05 \x03(\x0b\x32\x0f.xla.OpSharding\x12\"\n\x1areplicate_on_last_tile_dim\x18\x06 \x01(\x08\x12!\n\x08metadata\x18\x07 \x03(\x0b\x32\x0f.xla.OpMetadata\x12,\n\x0elast_tile_dims\x18\x08 \x03(\x0e\x32\x14.xla.OpSharding.Type\x12\x19\n\x11iota_reshape_dims\x18\t \x03(\x03\x12\x1b\n\x13iota_transpose_perm\x18\n \x03(\x05\x12\x16\n\x0eis_shard_group\x18\x0b \x01(\x08\x12\x16\n\x0eshard_group_id\x18\x0c \x01(\x03\x12\x38\n\x10shard_group_type\x18\r \x01(\x0e\x32\x1e.xla.OpSharding.ShardGroupType\x12\x34\n\x0enamed_sharding\x18\x0e \x01(\x0b\x32\x17.xla.NamedShardingProtoH\x00\x88\x01\x01\"a\n\x04Type\x12\x0e\n\nREPLICATED\x10\x00\x12\x0b\n\x07MAXIMAL\x10\x01\x12\t\n\x05TUPLE\x10\x02\x12\t\n\x05OTHER\x10\x03\x12\n\n\x06MANUAL\x10\x04\x12\x0b\n\x07UNKNOWN\x10\x05\x12\r\n\tUNREDUCED\x10\x06\"\"\n\x0eShardGroupType\x12\x06\n\x02\x41S\x10\x00\x12\x08\n\x04LIKE\x10\x01\x42\x11\n\x0f_named_sharding\"#\n\x0cReplicaGroup\x12\x13\n\x0breplica_ids\x18\x01 \x03(\x03\"^\n\x1dMeshAxesReplicaGroupListProto\x12\x1c\n\x04mesh\x18\x01 \x01(\x0b\x32\x0e.xla.MeshProto\x12\x1f\n\x04\x61xes\x18\x02 \x03(\x0b\x32\x11.xla.AxisRefProto\"\x8e\x01\n\x19IotaReplicaGroupListProto\x12\x1a\n\x12num_replica_groups\x18\x01 \x01(\x03\x12\x1d\n\x15num_devices_per_group\x18\x02 \x01(\x03\x12\x19\n\x11iota_reshape_dims\x18\x03 \x03(\x03\x12\x1b\n\x13iota_transpose_perm\x18\x04 \x03(\x05\"\x87\x01\n\x19\x43ollectiveDeviceListProto\x12)\n\x0ereplica_groups\x18\x01 \x03(\x0b\x32\x11.xla.ReplicaGroup\x12?\n\x17iota_replica_group_list\x18\x02 \x01(\x0b\x32\x1e.xla.IotaReplicaGroupListProto\".\n\x0cSourceTarget\x12\x0e\n\x06source\x18\x01 \x01(\x03\x12\x0e\n\x06target\x18\x02 \x01(\x03\"\xd0\x01\n\x0eResultAccuracy\x12(\n\x04mode\x18\x01 \x01(\x0e\x32\x18.xla.ResultAccuracy.ModeH\x00\x12\x32\n\ttolerance\x18\x02 \x01(\x0b\x32\x1d.xla.ResultAccuracy.ToleranceH\x00\x1a\x35\n\tTolerance\x12\x0c\n\x04\x61tol\x18\x01 \x01(\x01\x12\x0c\n\x04rtol\x18\x02 \x01(\x01\x12\x0c\n\x04ulps\x18\x03 \x01(\x03\" \n\x04Mode\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x0b\n\x07HIGHEST\x10\x01\x42\x07\n\x05specs\"\xc4\x04\n\x0fPrecisionConfig\x12\x39\n\x11operand_precision\x18\x01 \x03(\x0e\x32\x1e.xla.PrecisionConfig.Precision\x12\x31\n\talgorithm\x18\x02 \x01(\x0e\x32\x1e.xla.PrecisionConfig.Algorithm\"/\n\tPrecision\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x08\n\x04HIGH\x10\x01\x12\x0b\n\x07HIGHEST\x10\x02\"\x91\x03\n\tAlgorithm\x12\r\n\tALG_UNSET\x10\x00\x12\x1d\n\x19\x41LG_DOT_ANY_F8_ANY_F8_F32\x10\x01\x12(\n$ALG_DOT_ANY_F8_ANY_F8_F32_FAST_ACCUM\x10\x02\x12\x17\n\x13\x41LG_DOT_F16_F16_F16\x10\x03\x12\x17\n\x13\x41LG_DOT_F16_F16_F32\x10\x04\x12\x1a\n\x16\x41LG_DOT_BF16_BF16_BF16\x10\x05\x12\x19\n\x15\x41LG_DOT_BF16_BF16_F32\x10\x06\x12\x1c\n\x18\x41LG_DOT_BF16_BF16_F32_X3\x10\x07\x12\x1c\n\x18\x41LG_DOT_BF16_BF16_F32_X6\x10\x08\x12\x19\n\x15\x41LG_DOT_TF32_TF32_F32\x10\t\x12\x1c\n\x18\x41LG_DOT_TF32_TF32_F32_X3\x10\n\x12\x17\n\x13\x41LG_DOT_F32_F32_F32\x10\x0b\x12\x17\n\x13\x41LG_DOT_F64_F64_F64\x10\x0c\x12\x1c\n\x18\x41LG_DOT_BF16_BF16_F32_X9\x10\r\":\n\x14ParameterReplication\x12\"\n\x1areplicated_at_leaf_buffers\x18\x01 \x03(\x08\"\x99\x03\n\x16WhileLoopBackendConfig\x12\x44\n\x10known_trip_count\x18\x01 \x01(\x0b\x32*.xla.WhileLoopBackendConfig.KnownTripCount\x12\x42\n\x0fknown_init_step\x18\x02 \x01(\x0b\x32).xla.WhileLoopBackendConfig.KnownInitStep\x12T\n\x18known_induction_variable\x18\x03 \x01(\x0b\x32\x32.xla.WhileLoopBackendConfig.KnownInductionVariable\x12&\n\x1e\x64ynamic_variable_tuple_indices\x18\x04 \x03(\x03\x1a+\n\rKnownInitStep\x12\x0c\n\x04init\x18\x01 \x01(\x03\x12\x0c\n\x04step\x18\x02 \x01(\x03\x1a\x1b\n\x0eKnownTripCount\x12\t\n\x01n\x18\x01 \x01(\x03\x1a-\n\x16KnownInductionVariable\x12\x13\n\x0btuple_index\x18\x01 \x01(\x03\"g\n\x15OutputOperandAliasing\x12\x1a\n\x12output_shape_index\x18\x01 \x03(\x03\x12\x15\n\roperand_index\x18\x02 \x01(\x03\x12\x1b\n\x13operand_shape_index\x18\x03 \x03(\x03\"C\n\x12OriginalArrayProto\x12\x18\n\x10instruction_name\x18\x01 \x01(\t\x12\x13\n\x0bshape_index\x18\x02 \x03(\x03\"a\n\x19OriginalValueElementProto\x12\x13\n\x0bshape_index\x18\x01 \x03(\x03\x12/\n\x0eoriginal_array\x18\x02 \x01(\x0b\x32\x17.xla.OriginalArrayProto\"a\n\x12OriginalValueProto\x12\x30\n\x08\x65lements\x18\x01 \x03(\x0b\x32\x1e.xla.OriginalValueElementProto\x12\x19\n\x11is_synthetic_call\x18\x02 \x01(\x08\"\xa1\x01\n\x12GemmPerfTableEntry\x12\t\n\x01\x62\x18\x01 \x01(\x03\x12\t\n\x01m\x18\x02 \x01(\x03\x12\t\n\x01n\x18\x03 \x01(\x03\x12\t\n\x01k\x18\x04 \x01(\x03\x12\x31\n\x05\x66lops\x18\x05 \x03(\x0b\x32\".xla.GemmPerfTableEntry.FlopsEntry\x1a,\n\nFlopsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x03:\x02\x38\x01\"D\n\x18GemmPerfTableEntryValues\x12(\n\x07\x65ntries\x18\x01 \x03(\x0b\x32\x17.xla.GemmPerfTableEntry\"\x90\x01\n\rGemmPerfTable\x12\x30\n\x07\x65ntries\x18\x01 \x03(\x0b\x32\x1f.xla.GemmPerfTable.EntriesEntry\x1aM\n\x0c\x45ntriesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x1d.xla.GemmPerfTableEntryValues:\x02\x38\x01*\x98\x03\n\rPrimitiveType\x12\x1a\n\x16PRIMITIVE_TYPE_INVALID\x10\x00\x12\x08\n\x04PRED\x10\x01\x12\x06\n\x02S1\x10\x1e\x12\x06\n\x02S2\x10\x1a\x12\x06\n\x02S4\x10\x15\x12\x06\n\x02S8\x10\x02\x12\x07\n\x03S16\x10\x03\x12\x07\n\x03S32\x10\x04\x12\x07\n\x03S64\x10\x05\x12\x06\n\x02U1\x10\x1f\x12\x06\n\x02U2\x10\x1b\x12\x06\n\x02U4\x10\x16\x12\x06\n\x02U8\x10\x06\x12\x07\n\x03U16\x10\x07\x12\x07\n\x03U32\x10\x08\x12\x07\n\x03U64\x10\t\x12\x07\n\x03\x46\x31\x36\x10\n\x12\x07\n\x03\x46\x33\x32\x10\x0b\x12\x08\n\x04\x42\x46\x31\x36\x10\x10\x12\x07\n\x03\x46\x36\x34\x10\x0c\x12\n\n\x06\x46\x38\x45\x35M2\x10\x13\x12\n\n\x06\x46\x38\x45\x34M3\x10\x1c\x12\x0c\n\x08\x46\x38\x45\x34M3FN\x10\x14\x12\x11\n\rF8E4M3B11FNUZ\x10\x17\x12\n\n\x06\x46\x38\x45\x33M4\x10\x1d\x12\x0e\n\nF8E5M2FNUZ\x10\x18\x12\x0e\n\nF8E4M3FNUZ\x10\x19\x12\x0c\n\x08\x46\x34\x45\x32M1FN\x10 \x12\r\n\tF8E8M0FNU\x10!\x12\x07\n\x03\x43\x36\x34\x10\x0f\x12\x08\n\x04\x43\x31\x32\x38\x10\x12\x12\t\n\x05TUPLE\x10\r\x12\x0f\n\x0bOPAQUE_TYPE\x10\x0e\x12\t\n\x05TOKEN\x10\x11\x12\n\n\x06\x42UFFER\x10\"*\x8c\x01\n\x0f\x41syncStreamKind\x12 \n\x1c\x41SYNC_STREAM_KIND_COLLECTIVE\x10\x00\x12\x1a\n\x16\x41SYNC_STREAM_KIND_P2P0\x10\x01\x12\x1a\n\x16\x41SYNC_STREAM_KIND_P2P1\x10\x02\x12\x1f\n\x1b\x41SYNC_STREAM_KIND_MEMCPYP2P\x10\x03*\xd6\x01\n\x15\x43ollectiveOpGroupMode\x12*\n&COLLECTIVE_OP_GROUP_MODE_CROSS_REPLICA\x10\x00\x12,\n(COLLECTIVE_OP_GROUP_MODE_CROSS_PARTITION\x10\x01\x12\x38\n4COLLECTIVE_OP_GROUP_MODE_CROSS_REPLICA_AND_PARTITION\x10\x02\x12)\n%COLLECTIVE_OP_GROUP_MODE_FLATTENED_ID\x10\x03*^\n\x0c\x44imLevelType\x12\r\n\tDIM_DENSE\x10\x00\x12\x12\n\x0e\x44IM_COMPRESSED\x10\x01\x12\x11\n\rDIM_SINGLETON\x10\x02\x12\x18\n\x14\x44IM_LOOSE_COMPRESSED\x10\x03*=\n\x0bProfileType\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06WINDOW\x10\x01\x12\x08\n\x04\x46LAG\x10\x02\x12\x0b\n\x07INTEGER\x10\x03*j\n\rProfileSource\x12!\n\x1dPROFILE_SOURCE_UNKNOWN_SOURCE\x10\x00\x12\x1b\n\x17PROFILE_SOURCE_EMBEDDED\x10\x01\x12\x19\n\x15PROFILE_SOURCE_REMOTE\x10\x02*\x85\x01\n\x10\x43ompilationEvent\x12#\n\x1f\x43OMPILATION_EVENT_UNKNOWN_EVENT\x10\x00\x12\'\n#COMPILATION_EVENT_FIRST_COMPILATION\x10\x01\x12#\n\x1f\x43OMPILATION_EVENT_RECOMPILATION\x10\x02*\xb2\x02\n\x19ProfileGenerationStrategy\x12\'\n#PROFILE_GENERATION_STRATEGY_UNKNOWN\x10\x00\x12\"\n\x1ePROFILE_GENERATION_STRATEGY_GA\x10\x01\x12%\n!PROFILE_GENERATION_STRATEGY_FANTA\x10\x02\x12#\n\x1fPROFILE_GENERATION_STRATEGY_CFO\x10\x03\x12*\n&PROFILE_GENERATION_STRATEGY_EXHAUSTIVE\x10\x04\x12\'\n#PROFILE_GENERATION_STRATEGY_LCM_GNN\x10\x05\x12\'\n#PROFILE_GENERATION_STRATEGY_LCM_MOE\x10\x06*G\n\x0bPaddingType\x12\x13\n\x0fPADDING_INVALID\x10\x00\x12\x11\n\rPADDING_VALID\x10\x01\x12\x10\n\x0cPADDING_SAME\x10\x02*1\n\x07\x46\x66tType\x12\x07\n\x03\x46\x46T\x10\x00\x12\x08\n\x04IFFT\x10\x01\x12\x08\n\x04RFFT\x10\x02\x12\t\n\x05IRFFT\x10\x03*F\n\x12RandomDistribution\x12\x0f\n\x0bRNG_INVALID\x10\x00\x12\x0f\n\x0bRNG_UNIFORM\x10\x01\x12\x0e\n\nRNG_NORMAL\x10\x02*E\n\x0fRandomAlgorithm\x12\x0f\n\x0bRNG_DEFAULT\x10\x00\x12\x11\n\rRNG_THREE_FRY\x10\x01\x12\x0e\n\nRNG_PHILOX\x10\x02\x42\x03\xf8\x01\x01\x62\x06proto3') _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'xla.xla_data_pb2', _globals) if not _descriptor._USE_C_DESCRIPTORS: _globals['DESCRIPTOR']._loaded_options = None _globals['DESCRIPTOR']._serialized_options = b'\370\001\001' _globals['_OPMETADATA'].fields_by_name['profile_type']._loaded_options = None _globals['_OPMETADATA'].fields_by_name['profile_type']._serialized_options = b'\030\001' _globals['_FRONTENDATTRIBUTES_MAPENTRY']._loaded_options = None _globals['_FRONTENDATTRIBUTES_MAPENTRY']._serialized_options = b'8\001' _globals['_GEMMPERFTABLEENTRY_FLOPSENTRY']._loaded_options = None _globals['_GEMMPERFTABLEENTRY_FLOPSENTRY']._serialized_options = b'8\001' _globals['_GEMMPERFTABLE_ENTRIESENTRY']._loaded_options = None _globals['_GEMMPERFTABLE_ENTRIESENTRY']._serialized_options = b'8\001' _globals['_PRIMITIVETYPE']._serialized_start=8976 _globals['_PRIMITIVETYPE']._serialized_end=9384 _globals['_ASYNCSTREAMKIND']._serialized_start=9387 _globals['_ASYNCSTREAMKIND']._serialized_end=9527 _globals['_COLLECTIVEOPGROUPMODE']._serialized_start=9530 _globals['_COLLECTIVEOPGROUPMODE']._serialized_end=9744 _globals['_DIMLEVELTYPE']._serialized_start=9746 _globals['_DIMLEVELTYPE']._serialized_end=9840 _globals['_PROFILETYPE']._serialized_start=9842 _globals['_PROFILETYPE']._serialized_end=9903 _globals['_PROFILESOURCE']._serialized_start=9905 _globals['_PROFILESOURCE']._serialized_end=10011 _globals['_COMPILATIONEVENT']._serialized_start=10014 _globals['_COMPILATIONEVENT']._serialized_end=10147 _globals['_PROFILEGENERATIONSTRATEGY']._serialized_start=10150 _globals['_PROFILEGENERATIONSTRATEGY']._serialized_end=10456 _globals['_PADDINGTYPE']._serialized_start=10458 _globals['_PADDINGTYPE']._serialized_end=10529 _globals['_FFTTYPE']._serialized_start=10531 _globals['_FFTTYPE']._serialized_end=10580 _globals['_RANDOMDISTRIBUTION']._serialized_start=10582 _globals['_RANDOMDISTRIBUTION']._serialized_end=10652 _globals['_RANDOMALGORITHM']._serialized_start=10654 _globals['_RANDOMALGORITHM']._serialized_end=10723 _globals['_PADDINGCONFIG']._serialized_start=28 _globals['_PADDINGCONFIG']._serialized_end=211 _globals['_PADDINGCONFIG_PADDINGCONFIGDIMENSION']._serialized_start=108 _globals['_PADDINGCONFIG_PADDINGCONFIGDIMENSION']._serialized_end=211 _globals['_TILEPROTO']._serialized_start=213 _globals['_TILEPROTO']._serialized_end=244 _globals['_SPLITCONFIGPROTO']._serialized_start=246 _globals['_SPLITCONFIGPROTO']._serialized_end=306 _globals['_LAYOUTPROTO']._serialized_start=309 _globals['_LAYOUTPROTO']._serialized_end=879 _globals['_SHAPEPROTO']._serialized_start=882 _globals['_SHAPEPROTO']._serialized_end=1071 _globals['_PROGRAMSHAPEPROTO']._serialized_start=1073 _globals['_PROGRAMSHAPEPROTO']._serialized_end=1187 _globals['_COMPUTATIONSTATS']._serialized_start=1189 _globals['_COMPUTATIONSTATS']._serialized_end=1257 _globals['_OPMETADATA']._serialized_start=1260 _globals['_OPMETADATA']._serialized_end=1999 _globals['_OPMETADATA_PROFILEINFO']._serialized_start=1683 _globals['_OPMETADATA_PROFILEINFO']._serialized_end=1925 _globals['_EXECUTIONPROFILE']._serialized_start=2002 _globals['_EXECUTIONPROFILE']._serialized_end=2258 _globals['_EXECUTIONHANDLE']._serialized_start=2260 _globals['_EXECUTIONHANDLE']._serialized_end=2293 _globals['_GLOBALDATAHANDLE']._serialized_start=2295 _globals['_GLOBALDATAHANDLE']._serialized_end=2329 _globals['_DEVICEHANDLE']._serialized_start=2331 _globals['_DEVICEHANDLE']._serialized_end=2383 _globals['_CHANNELHANDLE']._serialized_start=2386 _globals['_CHANNELHANDLE']._serialized_end=2566 _globals['_CHANNELHANDLE_CHANNELTYPE']._serialized_start=2465 _globals['_CHANNELHANDLE_CHANNELTYPE']._serialized_end=2566 _globals['_DEVICEASSIGNMENTPROTO']._serialized_start=2569 _globals['_DEVICEASSIGNMENTPROTO']._serialized_end=2766 _globals['_DEVICEASSIGNMENTPROTO_COMPUTATIONDEVICE']._serialized_start=2719 _globals['_DEVICEASSIGNMENTPROTO_COMPUTATIONDEVICE']._serialized_end=2766 _globals['_LITERALPROTO']._serialized_start=2769 _globals['_LITERALPROTO']._serialized_end=3346 _globals['_WINDOWDIMENSION']._serialized_start=3349 _globals['_WINDOWDIMENSION']._serialized_end=3512 _globals['_WINDOW']._serialized_start=3514 _globals['_WINDOW']._serialized_end=3564 _globals['_GATHERDIMENSIONNUMBERS']._serialized_start=3567 _globals['_GATHERDIMENSIONNUMBERS']._serialized_end=3761 _globals['_SCATTERDIMENSIONNUMBERS']._serialized_start=3764 _globals['_SCATTERDIMENSIONNUMBERS']._serialized_end=3979 _globals['_CONVOLUTIONDIMENSIONNUMBERS']._serialized_start=3982 _globals['_CONVOLUTIONDIMENSIONNUMBERS']._serialized_end=4326 _globals['_DOTDIMENSIONNUMBERS']._serialized_start=4329 _globals['_DOTDIMENSIONNUMBERS']._serialized_end=4482 _globals['_RAGGEDDOTDIMENSIONNUMBERS']._serialized_start=4485 _globals['_RAGGEDDOTDIMENSIONNUMBERS']._serialized_end=4630 _globals['_TRIANGULARSOLVEOPTIONS']._serialized_start=4633 _globals['_TRIANGULARSOLVEOPTIONS']._serialized_end=4856 _globals['_TRIANGULARSOLVEOPTIONS_TRANSPOSE']._serialized_start=4776 _globals['_TRIANGULARSOLVEOPTIONS_TRANSPOSE']._serialized_end=4856 _globals['_CHOLESKYOPTIONS']._serialized_start=4858 _globals['_CHOLESKYOPTIONS']._serialized_end=4890 _globals['_SORTOPTIONS']._serialized_start=4892 _globals['_SORTOPTIONS']._serialized_end=4925 _globals['_FRONTENDATTRIBUTES']._serialized_start=4927 _globals['_FRONTENDATTRIBUTES']._serialized_end=5038 _globals['_FRONTENDATTRIBUTES_MAPENTRY']._serialized_start=4996 _globals['_FRONTENDATTRIBUTES_MAPENTRY']._serialized_end=5038 _globals['_STATISTIC']._serialized_start=5040 _globals['_STATISTIC']._serialized_end=5088 _globals['_STATISTICSVIZ']._serialized_start=5090 _globals['_STATISTICSVIZ']._serialized_end=5174 _globals['_MESHPROTO']._serialized_start=5176 _globals['_MESHPROTO']._serialized_end=5286 _globals['_MESHPROTO_MESHAXIS']._serialized_start=5248 _globals['_MESHPROTO_MESHAXIS']._serialized_end=5286 _globals['_AXISREFPROTO']._serialized_start=5289 _globals['_AXISREFPROTO']._serialized_end=5444 _globals['_AXISREFPROTO_SUBAXIS']._serialized_start=5385 _globals['_AXISREFPROTO_SUBAXIS']._serialized_end=5426 _globals['_NAMEDSHARDINGPROTO']._serialized_start=5447 _globals['_NAMEDSHARDINGPROTO']._serialized_end=5804 _globals['_NAMEDSHARDINGPROTO_DIMENSIONSHARDING']._serialized_start=5727 _globals['_NAMEDSHARDINGPROTO_DIMENSIONSHARDING']._serialized_end=5798 _globals['_OPSHARDING']._serialized_start=5807 _globals['_OPSHARDING']._serialized_end=6490 _globals['_OPSHARDING_TYPE']._serialized_start=6338 _globals['_OPSHARDING_TYPE']._serialized_end=6435 _globals['_OPSHARDING_SHARDGROUPTYPE']._serialized_start=6437 _globals['_OPSHARDING_SHARDGROUPTYPE']._serialized_end=6471 _globals['_REPLICAGROUP']._serialized_start=6492 _globals['_REPLICAGROUP']._serialized_end=6527 _globals['_MESHAXESREPLICAGROUPLISTPROTO']._serialized_start=6529 _globals['_MESHAXESREPLICAGROUPLISTPROTO']._serialized_end=6623 _globals['_IOTAREPLICAGROUPLISTPROTO']._serialized_start=6626 _globals['_IOTAREPLICAGROUPLISTPROTO']._serialized_end=6768 _globals['_COLLECTIVEDEVICELISTPROTO']._serialized_start=6771 _globals['_COLLECTIVEDEVICELISTPROTO']._serialized_end=6906 _globals['_SOURCETARGET']._serialized_start=6908 _globals['_SOURCETARGET']._serialized_end=6954 _globals['_RESULTACCURACY']._serialized_start=6957 _globals['_RESULTACCURACY']._serialized_end=7165 _globals['_RESULTACCURACY_TOLERANCE']._serialized_start=7069 _globals['_RESULTACCURACY_TOLERANCE']._serialized_end=7122 _globals['_RESULTACCURACY_MODE']._serialized_start=7124 _globals['_RESULTACCURACY_MODE']._serialized_end=7156 _globals['_PRECISIONCONFIG']._serialized_start=7168 _globals['_PRECISIONCONFIG']._serialized_end=7748 _globals['_PRECISIONCONFIG_PRECISION']._serialized_start=7297 _globals['_PRECISIONCONFIG_PRECISION']._serialized_end=7344 _globals['_PRECISIONCONFIG_ALGORITHM']._serialized_start=7347 _globals['_PRECISIONCONFIG_ALGORITHM']._serialized_end=7748 _globals['_PARAMETERREPLICATION']._serialized_start=7750 _globals['_PARAMETERREPLICATION']._serialized_end=7808 _globals['_WHILELOOPBACKENDCONFIG']._serialized_start=7811 _globals['_WHILELOOPBACKENDCONFIG']._serialized_end=8220 _globals['_WHILELOOPBACKENDCONFIG_KNOWNINITSTEP']._serialized_start=8101 _globals['_WHILELOOPBACKENDCONFIG_KNOWNINITSTEP']._serialized_end=8144 _globals['_WHILELOOPBACKENDCONFIG_KNOWNTRIPCOUNT']._serialized_start=8146 _globals['_WHILELOOPBACKENDCONFIG_KNOWNTRIPCOUNT']._serialized_end=8173 _globals['_WHILELOOPBACKENDCONFIG_KNOWNINDUCTIONVARIABLE']._serialized_start=8175 _globals['_WHILELOOPBACKENDCONFIG_KNOWNINDUCTIONVARIABLE']._serialized_end=8220 _globals['_OUTPUTOPERANDALIASING']._serialized_start=8222 _globals['_OUTPUTOPERANDALIASING']._serialized_end=8325 _globals['_ORIGINALARRAYPROTO']._serialized_start=8327 _globals['_ORIGINALARRAYPROTO']._serialized_end=8394 _globals['_ORIGINALVALUEELEMENTPROTO']._serialized_start=8396 _globals['_ORIGINALVALUEELEMENTPROTO']._serialized_end=8493 _globals['_ORIGINALVALUEPROTO']._serialized_start=8495 _globals['_ORIGINALVALUEPROTO']._serialized_end=8592 _globals['_GEMMPERFTABLEENTRY']._serialized_start=8595 _globals['_GEMMPERFTABLEENTRY']._serialized_end=8756 _globals['_GEMMPERFTABLEENTRY_FLOPSENTRY']._serialized_start=8712 _globals['_GEMMPERFTABLEENTRY_FLOPSENTRY']._serialized_end=8756 _globals['_GEMMPERFTABLEENTRYVALUES']._serialized_start=8758 _globals['_GEMMPERFTABLEENTRYVALUES']._serialized_end=8826 _globals['_GEMMPERFTABLE']._serialized_start=8829 _globals['_GEMMPERFTABLE']._serialized_end=8973 _globals['_GEMMPERFTABLE_ENTRIESENTRY']._serialized_start=8896 _globals['_GEMMPERFTABLE_ENTRIESENTRY']._serialized_end=8973 # @@protoc_insertion_point(module_scope)