# -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # source: tensorflow/compiler/mlir/quantization/stablehlo/quantization_config.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'\nItensorflow/compiler/mlir/quantization/stablehlo/quantization_config.proto\x12\x16stablehlo.quantization\"\x1c\n\x0cTfRecordFile\x12\x0c\n\x04path\x18\x01 \x01(\t\"\x8e\x01\n\x1bRepresentativeDatasetConfig\x12\x39\n\ttf_record\x18\x01 \x01(\x0b\x32$.stablehlo.quantization.TfRecordFileH\x00\x12\x1a\n\rsignature_key\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\x06\n\x04\x66ileB\x10\n\x0e_signature_key\"\xc3\x01\n\x14StaticRangePtqPreset\x12T\n\x17representative_datasets\x18\x01 \x03(\x0b\x32\x33.stablehlo.quantization.RepresentativeDatasetConfig\x12/\n#enable_per_channel_quantized_weight\x18\x02 \x01(\x08\x42\x02\x18\x01\x12$\n\x1c\x65nable_full_int_quantization\x18\x03 \x01(\x08\"\x15\n\x13WeightOnlyPtqPreset\"\"\n\x12TfSavedModelConfig\x12\x0c\n\x04tags\x18\x01 \x03(\t\"v\n\x0ePipelineConfig\x12#\n\x16unpack_quantized_types\x18\x01 \x01(\x08H\x00\x88\x01\x01\x12$\n\x1cmerge_fusion_with_dequantize\x18\x02 \x01(\x08\x42\x19\n\x17_unpack_quantized_types\"\x1f\n\x0fQuantizableUnit\x12\x0c\n\x04name\x18\x01 \x01(\t\"\x87\x01\n\x12QuantizationResult\x12\x41\n\x10quantizable_unit\x18\x01 \x01(\x0b\x32\'.stablehlo.quantization.QuantizableUnit\x12.\n\x06method\x18\x02 \x01(\x0b\x32\x1e.stablehlo.quantization.Method\"R\n\x13QuantizationResults\x12;\n\x07results\x18\x01 \x03(\x0b\x32*.stablehlo.quantization.QuantizationResult\":\n\x12QuantizedDimension\x12\x16\n\tdimension\x18\x01 \x01(\x05H\x00\x88\x01\x01\x42\x0c\n\n_dimension\"\x0b\n\tPerTensor\"\x97\x01\n\rQuantizedType\x12\x45\n\x0f\x64imension_specs\x18\x01 \x01(\x0b\x32*.stablehlo.quantization.QuantizedDimensionH\x00\x12\x37\n\nper_tensor\x18\x02 \x01(\x0b\x32!.stablehlo.quantization.PerTensorH\x00\x42\x06\n\x04type\"\x10\n\x0eNoQuantization\"\xd3\x01\n\x0eStaticRangePtq\x12^\n\x15input_quantized_types\x18\x01 \x03(\x0b\x32?.stablehlo.quantization.StaticRangePtq.InputQuantizedTypesEntry\x1a\x61\n\x18InputQuantizedTypesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.stablehlo.quantization.QuantizedType:\x02\x38\x01\"\xd1\x01\n\rWeightOnlyPtq\x12]\n\x15input_quantized_types\x18\x01 \x03(\x0b\x32>.stablehlo.quantization.WeightOnlyPtq.InputQuantizedTypesEntry\x1a\x61\n\x18InputQuantizedTypesEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.stablehlo.quantization.QuantizedType:\x02\x38\x01\"(\n\x17\x46unctionNameMatcherSpec\x12\r\n\x05regex\x18\x01 \x01(\t\"U\n\x0bMatcherSpec\x12\x46\n\rfunction_name\x18\x01 \x01(\x0b\x32/.stablehlo.quantization.FunctionNameMatcherSpec\"\xdb\x01\n\x06Method\x12\x41\n\x0fno_quantization\x18\x01 \x01(\x0b\x32&.stablehlo.quantization.NoQuantizationH\x00\x12\x42\n\x10static_range_ptq\x18\x02 \x01(\x0b\x32&.stablehlo.quantization.StaticRangePtqH\x00\x12@\n\x0fweight_only_ptq\x18\x03 \x01(\x0b\x32%.stablehlo.quantization.WeightOnlyPtqH\x00\x42\x08\n\x06method\"x\n\x10QuantizationSpec\x12\x34\n\x07matcher\x18\x01 \x01(\x0b\x32#.stablehlo.quantization.MatcherSpec\x12.\n\x06method\x18\x02 \x01(\x0b\x32\x1e.stablehlo.quantization.Method\"L\n\x11QuantizationSpecs\x12\x37\n\x05specs\x18\x01 \x03(\x0b\x32(.stablehlo.quantization.QuantizationSpec\"\xaa\x02\n\x0e\x44\x65\x62uggerConfig\x12J\n\rdebugger_type\x18\x01 \x01(\x0e\x32\x33.stablehlo.quantization.DebuggerConfig.DebuggerType\x12#\n\x1bunquantized_dump_model_path\x18\x02 \x01(\t\x12\x14\n\x0clog_dir_path\x18\x03 \x01(\t\"\x90\x01\n\x0c\x44\x65\x62uggerType\x12\x1d\n\x19\x44\x45\x42UGGER_TYPE_UNSPECIFIED\x10\x00\x12\x1d\n\x19\x44\x45\x42UGGER_TYPE_WHOLE_MODEL\x10\x01\x12\x1f\n\x1b\x44\x45\x42UGGER_TYPE_INT_PER_LAYER\x10\x02\x12!\n\x1d\x44\x45\x42UGGER_TYPE_FLOAT_PER_LAYER\x10\x03\"\x8e\x06\n\x12\x43\x61librationOptions\x12X\n\x12\x63\x61libration_method\x18\x01 \x01(\x0e\x32<.stablehlo.quantization.CalibrationOptions.CalibrationMethod\x12`\n\x16\x63\x61libration_parameters\x18\x02 \x01(\x0b\x32@.stablehlo.quantization.CalibrationOptions.CalibrationParameters\x12T\n\x17representative_datasets\x18\x03 \x03(\x0b\x32\x33.stablehlo.quantization.RepresentativeDatasetConfig\x12\x1c\n\x14\x63\x61libration_data_dir\x18\x04 \x01(\t\x12)\n!force_regenerate_calibration_data\x18\x05 \x01(\x08\x1aY\n\x15\x43\x61librationParameters\x12\x10\n\x08num_bins\x18\x01 \x01(\x05\x12\x16\n\x0emin_percentile\x18\x02 \x01(\x02\x12\x16\n\x0emax_percentile\x18\x03 \x01(\x02\"\xc1\x02\n\x11\x43\x61librationMethod\x12\"\n\x1e\x43\x41LIBRATION_METHOD_UNSPECIFIED\x10\x00\x12\x1e\n\x1a\x43\x41LIBRATION_METHOD_MIN_MAX\x10\x01\x12&\n\"CALIBRATION_METHOD_AVERAGE_MIN_MAX\x10\x02\x12+\n\'CALIBRATION_METHOD_HISTOGRAM_PERCENTILE\x10\x03\x12/\n+CALIBRATION_METHOD_HISTOGRAM_MSE_BRUTEFORCE\x10\x04\x12\x32\n.CALIBRATION_METHOD_HISTOGRAM_MSE_MAX_FREQUENCY\x10\x05\x12.\n*CALIBRATION_METHOD_HISTOGRAM_MSE_SYMMETRIC\x10\x06\"\xbb\x04\n\x12QuantizationConfig\x12O\n\x17static_range_ptq_preset\x18\x01 \x01(\x0b\x32,.stablehlo.quantization.StaticRangePtqPresetH\x00\x12M\n\x16weight_only_ptq_preset\x18\x07 \x01(\x0b\x32+.stablehlo.quantization.WeightOnlyPtqPresetH\x00\x12\x42\n\x0etf_saved_model\x18\x02 \x01(\x0b\x32*.stablehlo.quantization.TfSavedModelConfig\x12?\n\x0fpipeline_config\x18\x03 \x01(\x0b\x32&.stablehlo.quantization.PipelineConfig\x12\x38\n\x05specs\x18\x04 \x01(\x0b\x32).stablehlo.quantization.QuantizationSpecs\x12?\n\x0f\x64\x65\x62ugger_config\x18\x05 \x01(\x0b\x32&.stablehlo.quantization.DebuggerConfig\x12G\n\x13\x63\x61libration_options\x18\x06 \x01(\x0b\x32*.stablehlo.quantization.CalibrationOptions\x12\x1d\n\x10report_file_path\x18\x08 \x01(\tH\x01\x88\x01\x01\x42\x08\n\x06presetB\x13\n\x11_report_file_pathB\x03\xf8\x01\x01\x62\x06proto3') _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals()) _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'tensorflow.compiler.mlir.quantization.stablehlo.quantization_config_pb2', globals()) if _descriptor._USE_C_DESCRIPTORS == False: DESCRIPTOR._options = None DESCRIPTOR._serialized_options = b'\370\001\001' _STATICRANGEPTQPRESET.fields_by_name['enable_per_channel_quantized_weight']._options = None _STATICRANGEPTQPRESET.fields_by_name['enable_per_channel_quantized_weight']._serialized_options = b'\030\001' _STATICRANGEPTQ_INPUTQUANTIZEDTYPESENTRY._options = None _STATICRANGEPTQ_INPUTQUANTIZEDTYPESENTRY._serialized_options = b'8\001' _WEIGHTONLYPTQ_INPUTQUANTIZEDTYPESENTRY._options = None _WEIGHTONLYPTQ_INPUTQUANTIZEDTYPESENTRY._serialized_options = b'8\001' _TFRECORDFILE._serialized_start=101 _TFRECORDFILE._serialized_end=129 _REPRESENTATIVEDATASETCONFIG._serialized_start=132 _REPRESENTATIVEDATASETCONFIG._serialized_end=274 _STATICRANGEPTQPRESET._serialized_start=277 _STATICRANGEPTQPRESET._serialized_end=472 _WEIGHTONLYPTQPRESET._serialized_start=474 _WEIGHTONLYPTQPRESET._serialized_end=495 _TFSAVEDMODELCONFIG._serialized_start=497 _TFSAVEDMODELCONFIG._serialized_end=531 _PIPELINECONFIG._serialized_start=533 _PIPELINECONFIG._serialized_end=651 _QUANTIZABLEUNIT._serialized_start=653 _QUANTIZABLEUNIT._serialized_end=684 _QUANTIZATIONRESULT._serialized_start=687 _QUANTIZATIONRESULT._serialized_end=822 _QUANTIZATIONRESULTS._serialized_start=824 _QUANTIZATIONRESULTS._serialized_end=906 _QUANTIZEDDIMENSION._serialized_start=908 _QUANTIZEDDIMENSION._serialized_end=966 _PERTENSOR._serialized_start=968 _PERTENSOR._serialized_end=979 _QUANTIZEDTYPE._serialized_start=982 _QUANTIZEDTYPE._serialized_end=1133 _NOQUANTIZATION._serialized_start=1135 _NOQUANTIZATION._serialized_end=1151 _STATICRANGEPTQ._serialized_start=1154 _STATICRANGEPTQ._serialized_end=1365 _STATICRANGEPTQ_INPUTQUANTIZEDTYPESENTRY._serialized_start=1268 _STATICRANGEPTQ_INPUTQUANTIZEDTYPESENTRY._serialized_end=1365 _WEIGHTONLYPTQ._serialized_start=1368 _WEIGHTONLYPTQ._serialized_end=1577 _WEIGHTONLYPTQ_INPUTQUANTIZEDTYPESENTRY._serialized_start=1268 _WEIGHTONLYPTQ_INPUTQUANTIZEDTYPESENTRY._serialized_end=1365 _FUNCTIONNAMEMATCHERSPEC._serialized_start=1579 _FUNCTIONNAMEMATCHERSPEC._serialized_end=1619 _MATCHERSPEC._serialized_start=1621 _MATCHERSPEC._serialized_end=1706 _METHOD._serialized_start=1709 _METHOD._serialized_end=1928 _QUANTIZATIONSPEC._serialized_start=1930 _QUANTIZATIONSPEC._serialized_end=2050 _QUANTIZATIONSPECS._serialized_start=2052 _QUANTIZATIONSPECS._serialized_end=2128 _DEBUGGERCONFIG._serialized_start=2131 _DEBUGGERCONFIG._serialized_end=2429 _DEBUGGERCONFIG_DEBUGGERTYPE._serialized_start=2285 _DEBUGGERCONFIG_DEBUGGERTYPE._serialized_end=2429 _CALIBRATIONOPTIONS._serialized_start=2432 _CALIBRATIONOPTIONS._serialized_end=3214 _CALIBRATIONOPTIONS_CALIBRATIONPARAMETERS._serialized_start=2801 _CALIBRATIONOPTIONS_CALIBRATIONPARAMETERS._serialized_end=2890 _CALIBRATIONOPTIONS_CALIBRATIONMETHOD._serialized_start=2893 _CALIBRATIONOPTIONS_CALIBRATIONMETHOD._serialized_end=3214 _QUANTIZATIONCONFIG._serialized_start=3217 _QUANTIZATIONCONFIG._serialized_end=3788 # @@protoc_insertion_point(module_scope)