Index

A

AcceleratorRegistry (class in neural_compressor.torch.utils.auto_accelerator) Accuracy (class in neural_compressor.metric.metric) (class in neural_compressor.objective) AccuracyCriterion (class in neural_compressor.config) acq_max() (in module neural_compressor.strategy.bayesian) Algorithm (class in neural_compressor.algorithm.algorithm) algorithm_registry() (in module neural_compressor.algorithm.algorithm) ALGORITHMS (class in neural_compressor.algorithm.algorithm) AlgorithmScheduler (class in neural_compressor.algorithm.algorithm) alias_param() (in module neural_compressor.utils.utility) AlignImageChannelTransform (class in neural_compressor.data.transforms.transform) apply_inlining() (in module neural_compressor.tensorflow.quantization.utils.utility) apply_single_pattern_pair() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)

Auto_Accelerator (class in neural_compressor.torch.utils.auto_accelerator) auto_detect_accelerator() (in module neural_compressor.torch.utils.auto_accelerator) AutoAlpha (class in neural_compressor.torch.algorithms.smooth_quant.utility) AutoMixedPrecisionTuneStrategy (class in neural_compressor.strategy.auto_mixed_precision) autoround_quantize_entry() (in module neural_compressor.torch.quantization.algorithm_entry) AutoRoundConfig (class in neural_compressor.torch.quantization.config) AutoRoundQuantizer (class in neural_compressor.torch.algorithms.autoround.autoround) autotune() (in module neural_compressor.tensorflow.quantization.autotune) (in module neural_compressor.torch.quantization.autotune) AutoTuneStrategy (class in neural_compressor.strategy.auto) awq_quantize_entry() (in module neural_compressor.torch.quantization.algorithm_entry) AWQConfig (class in neural_compressor.torch.quantization.config) AWQQuantizer (class in neural_compressor.torch.algorithms.weight_only.awq) axis (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)

B

BaseConfig (class in neural_compressor.common.base_config) BaseDataLoader (class in neural_compressor.tensorflow.utils.data) BaseMetric (class in neural_compressor.metric.metric) BaseModel (class in neural_compressor.model.base_model) (class in neural_compressor.tensorflow.utils.model_wrappers) BaseTransform (class in neural_compressor.data.transforms.transform) BasicTokenizer (class in neural_compressor.data.transforms.tokenization) BasicTuneStrategy (class in neural_compressor.strategy.basic) BatchSampler (class in neural_compressor.tensorflow.utils.data) BayesianOptimization (class in neural_compressor.strategy.bayesian) BayesianTuneStrategy (class in neural_compressor.strategy.bayesian)

BenchmarkConfig (class in neural_compressor.config) BF16Convert (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert) BiasCorrection (class in neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction) BilinearImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) BLEU (class in neural_compressor.metric.bleu) bleu_tokenize() (in module neural_compressor.metric.bleu) block_wise_calibration() (in module neural_compressor.torch.utils.block_wise) BlockFallbackTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) build_captured_dataloader() (in module neural_compressor.torch.algorithms.smooth_quant.utility) build_slave_faker_model() (in module neural_compressor.strategy.utils.utility) bypass_reshape() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)

C

cal_scale() (in module neural_compressor.torch.algorithms.smooth_quant.utility) calculate_md5() (in module neural_compressor.data.datasets.dataset) calculate_mse() (in module neural_compressor.utils.utility) Calibration (class in neural_compressor.torch.algorithms.smooth_quant.utility) call_counter() (in module neural_compressor.common.utils.utility) CallBacks (class in neural_compressor.training) can_pack_with_numba() (in module neural_compressor.torch.utils.environ) CaptureOutputToFile (class in neural_compressor.tensorflow.utils.utility) (class in neural_compressor.utils.utility) CastONNXTransform (class in neural_compressor.data.transforms.transform) CastPyTorchTransform (class in neural_compressor.data.transforms.transform) CastTFTransform (class in neural_compressor.data.transforms.transform) CenterCropTFTransform (class in neural_compressor.data.transforms.transform) CenterCropTransform (class in neural_compressor.data.transforms.transform) cfg_to_qconfig() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.static_quant.utility) change_config_to_hf_format() (in module neural_compressor.torch.algorithms.weight_only.save_load) check_cfg_and_qconfig() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.static_quant.utility) check_integrity() (in module neural_compressor.data.datasets.dataset) check_key_exist() (in module neural_compressor.utils.utility) check_model() (in module neural_compressor.utils.export.qlinear2qdq) checkpoint_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) CIFAR10 (class in neural_compressor.data.datasets.dataset) CIFAR100 (class in neural_compressor.data.datasets.dataset) ClassRegister (class in neural_compressor.strategy.utils.utility) clean_module_weight() (in module neural_compressor.torch.algorithms.layer_wise.utils) COCOEvalWrapper (class in neural_compressor.metric.coco_tools) COCOmAPv2 (class in neural_compressor.metric.metric) COCONpy (class in neural_compressor.data.datasets.coco_dataset) COCORaw (class in neural_compressor.data.datasets.coco_dataset) COCORecordDataset (class in neural_compressor.data.datasets.coco_dataset) COCOWrapper (class in neural_compressor.metric.coco_tools) collate_tf_preds() (in module neural_compressor.tensorflow.quantization.utils.utility) CollectTransform (class in neural_compressor.data.transforms.transform) combine_histogram() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) compare_label (neural_compressor.metric.metric.MAE attribute) (neural_compressor.metric.metric.MSE attribute)

compare_objects() (in module neural_compressor.utils.utility) ComposableConfig (class in neural_compressor.common.base_config) ComposeTransform (class in neural_compressor.data.transforms.transform) CompressionManager (class in neural_compressor.training) compute_bleu() (in module neural_compressor.metric.bleu_util) compute_sparsity() (in module neural_compressor.utils.utility) config_list (neural_compressor.common.base_config.ComposableConfig attribute) (neural_compressor.common.base_tuning.ConfigSet attribute) ConfigLoader (class in neural_compressor.common.base_tuning) ConfigRegistry (class in neural_compressor.common.base_config) ConfigSet (class in neural_compressor.common.base_tuning) ConservativeTuneStrategy (class in neural_compressor.strategy.conservative) construct_function_from_graph_def() (in module neural_compressor.tensorflow.quantization.utils.utility) convert() (in module neural_compressor.torch.algorithms.qat.quant_utils) (in module neural_compressor.torch.quantization.quantize) convert_by_vocab() (in module neural_compressor.data.transforms.tokenization) convert_examples_to_features() (in module neural_compressor.data.datasets.bert_dataset) (in module neural_compressor.data.transforms.transform) convert_model_with_mapping() (in module neural_compressor.torch.algorithms.qat.quant_utils) convert_to_unicode() (in module neural_compressor.data.transforms.tokenization) ConvertAddToBiasAddOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd) ConvertLayoutOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout) ConvertLeakyReluOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu) ConvertNanToRandom (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random) ConvertPlaceholderToConst (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const) CPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator) CpuInfo (class in neural_compressor.common.utils.utility) (class in neural_compressor.tensorflow.utils.utility) (class in neural_compressor.utils.utility) create_dataloader() (in module neural_compressor.utils.create_obj_from_config) create_dataset() (in module neural_compressor.utils.create_obj_from_config) create_eval_func() (in module neural_compressor.utils.create_obj_from_config) create_quant_spec_from_config() (in module neural_compressor.torch.algorithms.pt2e_quant.utility) create_train_func() (in module neural_compressor.utils.create_obj_from_config) create_xiq_quantizer_from_pt2e_config() (in module neural_compressor.torch.algorithms.pt2e_quant.utility) CropResizeTFTransform (class in neural_compressor.data.transforms.transform) CropResizeTransform (class in neural_compressor.data.transforms.transform) CropToBoundingBox (class in neural_compressor.data.transforms.transform) CUDA_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)

D

Dataset (class in neural_compressor.data.datasets.dataset) dataset (neural_compressor.metric.coco_tools.COCOWrapper attribute) dataset_registry() (in module neural_compressor.data.datasets.dataset) Datasets (class in neural_compressor.data.datasets.dataset) debug() (in module neural_compressor.utils.logger) deep_get() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) deep_set() (in module neural_compressor.utils.utility) default_collate() (in module neural_compressor.tensorflow.utils.data) Dequantize() (in module neural_compressor.utils.utility) dequantize_weight() (in module neural_compressor.utils.utility) DequantizeCastOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer) detect_device() (in module neural_compressor.torch.utils.utility) detect_processor_type_based_on_hw() (in module neural_compressor.common.utils.utility) detection_type (neural_compressor.metric.coco_tools.COCOWrapper attribute) device_synchronize() (in module neural_compressor.torch.utils.environ) DilatedContraction (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction) disable_random() (in module neural_compressor.tensorflow.utils.utility) DistillationConfig (class in neural_compressor.config) DotDict (class in neural_compressor.config) (class in neural_compressor.utils.utility)

dowload_hf_model() (in module neural_compressor.torch.utils.utility) download_url() (in module neural_compressor.data.datasets.dataset) DummyDataset (class in neural_compressor.data.datasets.dummy_dataset) (class in neural_compressor.data.datasets.dummy_dataset_v2) (class in neural_compressor.tensorflow.utils.data) DummyDatasetV2 (class in neural_compressor.tensorflow.utils.data) dump_class_attrs() (in module neural_compressor.utils.utility) dump_data_to_local() (in module neural_compressor.utils.utility) dump_elapsed_time() (in module neural_compressor.common.utils.utility) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) dump_model_op_stats() (in module neural_compressor.torch.algorithms.autoround.autoround) (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.static_quant.utility) (in module neural_compressor.torch.utils.utility) dump_table() (in module neural_compressor.utils.utility) dump_table_to_csv() (in module neural_compressor.utils.utility) dynamic_quant_export() (in module neural_compressor.utils.export.torch2onnx) DynamicQuantConfig (class in neural_compressor.torch.quantization.config)

E

ElemFormat (class in neural_compressor.torch.algorithms.mx_quant.utils) enough_memo_store_scale() (in module neural_compressor.torch.algorithms.smooth_quant.utility) equal_dicts() (in module neural_compressor.utils.utility) error() (in module neural_compressor.utils.logger) estimator_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) evaluate() (in module neural_compressor.metric.evaluate_squad) (in module neural_compressor.metric.f1) EvaluationFuncWrapper (class in neural_compressor.common.base_tuning) Evaluator (class in neural_compressor.common.base_tuning) exact_match_score() (in module neural_compressor.metric.evaluate_squad)

ExhaustiveTuneStrategy (class in neural_compressor.strategy.exhaustive) ExpandDimsOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer) export() (in module neural_compressor.torch.export.pt2e_export) export_compressed_model() (in module neural_compressor.utils.load_huggingface) export_hf2compressored_model() (in module neural_compressor.torch.export.export_hf) export_model_for_pt2e_quant() (in module neural_compressor.torch.export.pt2e_export) ExportConfig (class in neural_compressor.config) ExportSingleImageDetectionBoxesToCoco() (in module neural_compressor.metric.coco_tools) ExportSingleImageDetectionMasksToCoco() (in module neural_compressor.metric.coco_tools) ExportSingleImageGroundtruthToCoco() (in module neural_compressor.metric.coco_tools) extract_data_type() (in module neural_compressor.strategy.utils.utility)

F

F1 (class in neural_compressor.metric.metric) f1_score() (in module neural_compressor.metric.evaluate_squad) (in module neural_compressor.metric.f1) FakeAffineTensorQuantFunction (class in neural_compressor.torch.algorithms.weight_only.modules) FallbackTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) FashionMNIST (class in neural_compressor.data.datasets.dataset) FastBiasCorrection (class in neural_compressor.algorithm.fast_bias_correction) fatal() (in module neural_compressor.utils.logger) fault_tolerant_file() (in module neural_compressor.utils.utility) fetch_module() (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.utility) FetchWeightFromReshapeOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape) Filter (class in neural_compressor.data.filters.filter) filter_fn() (in module neural_compressor.torch.algorithms.weight_only.hqq.quantizer) filter_registry() (in module neural_compressor.data.filters.filter) FILTERS (class in neural_compressor.data.filters.filter) finalize_calibration() (in module neural_compressor.torch.quantization.quantize) find_all_layers() (in module neural_compressor.torch.algorithms.weight_only.gptq) find_layers() (in module neural_compressor.torch.algorithms.weight_only.gptq) find_layers_name() (in module neural_compressor.torch.algorithms.weight_only.gptq) find_matching_blocks() (in module neural_compressor.torch.utils.utility) fit() (in module neural_compressor.mix_precision) (in module neural_compressor.quantization) (in module neural_compressor.training) fix_ref_type_of_graph_def() (in module neural_compressor.tensorflow.quantization.utils.utility) fn (neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter.PatternPair attribute) FoldBatchNormNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm) Footprint (class in neural_compressor.objective) forward_wrapper() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.utility) fp8_entry() (in module neural_compressor.torch.quantization.algorithm_entry) FP8Config (class in neural_compressor.torch.quantization.config) framework_datasets (in module neural_compressor.data.datasets.dataset)

FreezeFakeQuantOpOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant) FreezeValueTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value) frozen_pb_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) FullTokenizer (class in neural_compressor.data.transforms.tokenization) FuseBiasAddAndAddOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add) FuseColumnWiseMulOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul) FuseConvRedundantDequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize) FuseConvRequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize) FuseConvWithMathOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math) FuseDecomposedBNOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) FuseDecomposedINOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) FuseGeluOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu) FuseLayerNormOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm) FuseMatMulRedundantDequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize) FuseMatMulRequantizeDequantizeNewAPITransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseMatMulRequantizeDequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseMatMulRequantizeNewAPITransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseMatMulRequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseNodeStartWithConcatV2 (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_concatv2) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_concatv2) FuseNodeStartWithConv2d (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_conv) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_conv) FuseNodeStartWithDeconv2d (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv) FuseNodeStartWithFusedBatchNormV3 (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_bn) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_bn) FuseNodeStartWithFusedInstanceNorm (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in) FuseNodeStartWithMatmul (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_matmul) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_matmul) FuseNodeStartWithPooling (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_pooling) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling) FusePadWithConv2DOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv) FusePadWithFP32Conv2DOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv) FuseTransposeReshapeOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose)

G

gen_bar_updater() (in module neural_compressor.data.datasets.dataset) GeneralTopK (class in neural_compressor.metric.metric) generate_activation_observer() (in module neural_compressor.torch.algorithms.static_quant.utility) generate_feed_dict() (in module neural_compressor.tensorflow.quantization.utils.utility) generate_xpu_qconfig() (in module neural_compressor.torch.algorithms.static_quant.utility) GenerateGraphWithQDQPattern (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern) get_absorb_layers() (in module neural_compressor.torch.algorithms.weight_only.utility) get_accelerator() (in module neural_compressor.torch.utils.environ) get_adaptor_name() (in module neural_compressor.strategy.utils.utility) get_algorithm() (in module neural_compressor.utils.create_obj_from_config) get_all_config_set() (in module neural_compressor.tensorflow.quantization.autotune) (in module neural_compressor.torch.quantization.autotune) get_all_config_set_from_config_registry() (in module neural_compressor.common.base_config) get_all_fp32_data() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) get_all_registered_configs() (in module neural_compressor.tensorflow.keras.quantization.config) (in module neural_compressor.torch.quantization.config) get_block_names() (in module neural_compressor.torch.utils.utility) get_block_prefix() (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.block_wise) get_children() (in module neural_compressor.torch.algorithms.layer_wise.utils) get_const_dim_count() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) get_dataloader() (in module neural_compressor.torch.algorithms.autoround.autoround) get_default_AutoRound_config() (in module neural_compressor.torch.quantization.config) get_default_awq_config() (in module neural_compressor.torch.quantization.config) get_default_double_quant_config() (in module neural_compressor.torch.quantization.config) get_default_dynamic_config() (in module neural_compressor.torch.quantization.config) get_default_fp8_config() (in module neural_compressor.torch.quantization.config) get_default_fp8_config_set() (in module neural_compressor.torch.quantization.config) get_default_gptq_config() (in module neural_compressor.torch.quantization.config) get_default_hqq_config() (in module neural_compressor.torch.quantization.config) get_default_mixed_precision_config() (in module neural_compressor.torch.quantization.config) get_default_mixed_precision_config_set() (in module neural_compressor.torch.quantization.config) get_default_mx_config() (in module neural_compressor.torch.quantization.config) get_default_qat_module_mappings() (in module neural_compressor.torch.quantization.config) get_default_rtn_config() (in module neural_compressor.torch.quantization.config) get_default_sq_config() (in module neural_compressor.tensorflow.quantization.config) (in module neural_compressor.torch.quantization.config) get_default_static_config() (in module neural_compressor.torch.quantization.config) get_default_static_quant_config() (in module neural_compressor.tensorflow.keras.quantization.config) (in module neural_compressor.tensorflow.quantization.config) get_default_teq_config() (in module neural_compressor.torch.quantization.config) get_depth() (in module neural_compressor.torch.algorithms.static_quant.utility) get_dict_at_depth() (in module neural_compressor.torch.algorithms.static_quant.utility) get_double_quant_config_dict() (in module neural_compressor.torch.utils.utility) get_element_under_depth() (in module neural_compressor.torch.algorithms.static_quant.utility) get_enum_from_format() (in module neural_compressor.torch.utils.utility) get_filter_fn() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter) get_final_text() (in module neural_compressor.data.transforms.transform) get_func_from_config() (in module neural_compressor.utils.create_obj_from_config) get_graph_def() (in module neural_compressor.tensorflow.quantization.utils.utility) get_half_precision_node_set() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter) get_input_output_node_names() (in module neural_compressor.tensorflow.quantization.utils.utility) get_ipex_version() (in module neural_compressor.torch.utils.environ) get_layer_names_in_block() (in module neural_compressor.torch.utils.utility) get_metrics() (in module neural_compressor.utils.create_obj_from_config) get_mllm_dataloader() (in module neural_compressor.torch.algorithms.autoround.autoround) get_model_device() (in module neural_compressor.torch.utils.utility) get_model_fwk_name() (in module neural_compressor.model.model) get_model_info() (in module neural_compressor.torch.utils.utility) get_model_input_shape() (in module neural_compressor.tensorflow.quantization.utils.utility)

get_model_type() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) get_module() (in module neural_compressor.torch.algorithms.layer_wise.utils) (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.utility) get_module_input_output() (in module neural_compressor.torch.algorithms.weight_only.utility) get_multimodal_block_names() (in module neural_compressor.torch.utils.utility) get_named_children() (in module neural_compressor.torch.algorithms.layer_wise.utils) get_node_mapping() (in module neural_compressor.utils.export.torch2onnx) get_non_persistent_buffers() (in module neural_compressor.torch.utils.utility) get_number_of_sockets() (in module neural_compressor.utils.utility) get_op_list() (in module neural_compressor.utils.utility) get_parent() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) get_postprocess() (in module neural_compressor.utils.create_obj_from_config) get_preprocess() (in module neural_compressor.utils.create_obj_from_config) get_processor_type_from_user_config() (in module neural_compressor.torch.utils.utility) get_quant_config() (in module neural_compressor.torch.algorithms.qat.quant_utils) get_quant_config_with_scheme() (in module neural_compressor.torch.algorithms.qat.quant_utils) get_quantizable_onnx_ops() (in module neural_compressor.utils.export.torch2onnx) get_quantizable_ops_from_cfgs() (in module neural_compressor.torch.algorithms.static_quant.utility) get_quantizable_ops_recursively() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.static_quant.utility) get_quantization_format() (in module neural_compressor.torch.algorithms.qat.quant_utils) get_quantizer() (in module neural_compressor.torch.utils.utility) get_rtn_double_quant_config_set() (in module neural_compressor.torch.quantization.autotune) get_size() (in module neural_compressor.utils.utility) get_super_module_by_name() (in module neural_compressor.torch.algorithms.layer_wise.utils) get_tensor_by_name() (in module neural_compressor.tensorflow.quantization.utils.utility) get_tensor_histogram() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) get_tensors_info() (in module neural_compressor.utils.utility) get_tf_model_type() (in module neural_compressor.tensorflow.utils.model_wrappers) get_torch_version() (in module neural_compressor.torch.utils.environ) get_torchvision_map() (in module neural_compressor.data.transforms.transform) get_tuning_history() (in module neural_compressor.utils.utility) get_unquantized_node_set() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter) get_used_cpu_mem_MB() (in module neural_compressor.torch.utils.environ) get_used_hpu_mem_MB() (in module neural_compressor.torch.utils.environ) get_weights_details() (in module neural_compressor.utils.utility) get_woq_tuning_config() (in module neural_compressor.torch.quantization.config) get_workspace() (in module neural_compressor.common.utils.utility) GLOBAL_STATE (class in neural_compressor.utils.utility) GPTQ (class in neural_compressor.torch.algorithms.weight_only.gptq) gptq_entry() (in module neural_compressor.torch.quantization.algorithm_entry) GPTQConfig (class in neural_compressor.torch.quantization.config) GPTQuantizer (class in neural_compressor.torch.algorithms.weight_only.gptq) graph_def_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) graph_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) GraphAnalyzer (class in neural_compressor.tensorflow.quantization.utils.graph_util) GraphConverter (class in neural_compressor.tensorflow.quantization.utils.graph_converter) GraphCseOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer) GraphFoldConstantOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant) GraphRewriterBase (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base) GraphRewriterHelper (class in neural_compressor.tensorflow.quantization.utils.graph_util) GraphTrace (class in neural_compressor.torch.algorithms.smooth_quant.utility) (class in neural_compressor.torch.algorithms.weight_only.utility) GraphTransformBase (class in neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base) GrapplerOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass) group_size (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)

H

HalfPrecisionConverter (class in neural_compressor.torch.algorithms.mixed_precision.half_precision_convert) HalfPrecisionModuleWrapper (class in neural_compressor.torch.algorithms.mixed_precision.module_wrappers) HAWQ_V2TuneStrategy (class in neural_compressor.strategy.hawq_v2) HPOConfig (class in neural_compressor.config) HPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator) HPUWeightOnlyLinear (class in neural_compressor.torch.algorithms.weight_only.modules) hqq_entry() (in module neural_compressor.torch.quantization.algorithm_entry)

HQQConfig (class in neural_compressor.torch.quantization.config) HQQLinear (class in neural_compressor.torch.algorithms.weight_only.hqq.core) HQQModuleConfig (class in neural_compressor.torch.algorithms.weight_only.hqq.config) HQQTensorHandle (class in neural_compressor.torch.algorithms.weight_only.hqq.core) HQQuantizer (class in neural_compressor.torch.algorithms.weight_only.hqq.quantizer) hybrid_gptq_entry() (in module neural_compressor.torch.quantization.algorithm_entry) HybridGPTQConfig (class in neural_compressor.torch.quantization.config)

I

ImageFolder (class in neural_compressor.data.datasets.dataset) ImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) INCAcceleratorType (class in neural_compressor.torch.utils.auto_accelerator) INCWeightOnlyLinear (class in neural_compressor.torch.algorithms.weight_only.modules) IndexFetcher (class in neural_compressor.tensorflow.utils.data) info() (in module neural_compressor.utils.logger) init_tuning() (in module neural_compressor.common.base_tuning) initial_tuning_cfg_with_quant_mode() (in module neural_compressor.strategy.utils.tuning_space) initialize_int8_avgpool() (in module neural_compressor.tensorflow.keras.layers.pool2d) initialize_int8_conv2d() (in module neural_compressor.tensorflow.keras.layers.conv2d) initialize_int8_dense() (in module neural_compressor.tensorflow.keras.layers.dense) initialize_int8_depthwise_conv2d() (in module neural_compressor.tensorflow.keras.layers.depthwise_conv2d) initialize_int8_maxpool() (in module neural_compressor.tensorflow.keras.layers.pool2d) initialize_int8_separable_conv2d() (in module neural_compressor.tensorflow.keras.layers.separable_conv2d) InjectDummyBiasAddOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd) InputFeatures (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.data.transforms.transform) InsertLogging (class in neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging) InsertPrintMinMaxNode (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node) INT8StaticQuantConfig (class in neural_compressor.torch.quantization.config) IntermediateLayersKnowledgeDistillationLossConfig (class in neural_compressor.config)

IPEXModel (class in neural_compressor.model.torch_model) is_ckpt_format() (in module neural_compressor.tensorflow.quantization.utils.utility) is_hpex_available() (in module neural_compressor.torch.utils.environ) is_hpex_support_g_idx() (in module neural_compressor.torch.utils.environ) is_hpu_available() (in module neural_compressor.torch.utils.environ) is_int8_model() (in module neural_compressor.utils.pytorch) is_ipex_available() (in module neural_compressor.torch.utils.environ) is_ipex_imported() (in module neural_compressor.torch.utils.environ) is_leaf() (in module neural_compressor.torch.algorithms.weight_only.gptq) is_numba_available() (in module neural_compressor.torch.utils.environ) is_optimum_available() (in module neural_compressor.torch.utils.environ) is_optimum_habana_available() (in module neural_compressor.torch.utils.environ) is_package_available() (in module neural_compressor.torch.utils.environ) is_quantlinear() (in module neural_compressor.torch.algorithms.qat.quant_utils) is_saved_model_format() (in module neural_compressor.tensorflow.quantization.utils.utility) is_tbb_available() (in module neural_compressor.torch.utils.environ) is_transformers_imported() (in module neural_compressor.torch.utils.environ) IterableDataset (class in neural_compressor.data.datasets.dataset) IterableFetcher (class in neural_compressor.tensorflow.utils.data) IterableSampler (class in neural_compressor.tensorflow.utils.data) iterator_sess_run() (in module neural_compressor.tensorflow.quantization.utils.utility) itex_installed() (in module neural_compressor.tensorflow.utils.utility)

K

k (neural_compressor.metric.metric.GeneralTopK attribute) (neural_compressor.metric.metric.TensorflowTopK attribute) Keras (class in neural_compressor.config) keras_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) KerasAdaptor (class in neural_compressor.tensorflow.algorithms.static_quant.keras)

KerasConfigConverter (class in neural_compressor.tensorflow.algorithms.static_quant.keras) KerasModel (class in neural_compressor.model.keras_model) (class in neural_compressor.tensorflow.utils.model_wrappers) KerasQuery (class in neural_compressor.tensorflow.algorithms.static_quant.keras) KerasSurgery (class in neural_compressor.tensorflow.algorithms.static_quant.keras) KL_Divergence (class in neural_compressor.utils.kl_divergence) KnowledgeDistillationLossConfig (class in neural_compressor.config)

L

label_list (neural_compressor.metric.metric.Accuracy attribute) (neural_compressor.metric.metric.MAE attribute) (neural_compressor.metric.metric.MSE attribute) LabelBalanceCOCORawFilter (class in neural_compressor.data.filters.coco_filter) LabelBalanceCOCORecordFilter (class in neural_compressor.data.filters.coco_filter) labels (neural_compressor.metric.bleu.BLEU attribute) LabelShift (class in neural_compressor.data.transforms.imagenet_transform) LayerHistogramCollector (class in neural_compressor.utils.collect_layer_histogram) LazyImport (class in neural_compressor.common.utils.utility) (class in neural_compressor.utils.utility) load() (in module neural_compressor.torch.algorithms.layer_wise.load) (in module neural_compressor.torch.algorithms.pt2e_quant.save_load) (in module neural_compressor.torch.algorithms.static_quant.save_load) (in module neural_compressor.torch.algorithms.weight_only.save_load) (in module neural_compressor.torch.quantization.save_load_entry) (in module neural_compressor.utils.pytorch) load_and_cache_examples() (in module neural_compressor.data.datasets.bert_dataset) load_config_mapping() (in module neural_compressor.common.utils.save_load) load_data_from_pkl() (in module neural_compressor.utils.utility) load_empty_model() (in module neural_compressor.torch.utils.utility)

load_first_layer_only() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_layer_wise_quantized_model() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_model_from_shards_with_safetensors() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_module() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_non_persistent_buffers() (in module neural_compressor.torch.utils.utility) load_saved_model() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) load_tensor() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_tensor_from_safetensors() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_tensor_from_safetensors_shard() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_tensor_from_shard() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_value() (in module neural_compressor.torch.algorithms.layer_wise.utils) load_vocab() (in module neural_compressor.data.transforms.tokenization) load_weight_only() (in module neural_compressor.utils.pytorch) log() (in module neural_compressor.utils.logger) log_process() (in module neural_compressor.common.utils.utility) log_quantizable_layers_per_transformer() (in module neural_compressor.torch.algorithms.weight_only.gptq) Logger (class in neural_compressor.common.utils.logger) (class in neural_compressor.utils.logger) Loss (class in neural_compressor.metric.metric) LowerBitsSampler (class in neural_compressor.strategy.utils.tuning_sampler)

M

MAE (class in neural_compressor.metric.metric) Matmul (class in neural_compressor.torch.algorithms.weight_only.modules) MergeDuplicatedQDQOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq) MetaInfoChangingMemOpOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer) Metric (class in neural_compressor.metric.metric) metric_max_over_ground_truths() (in module neural_compressor.metric.evaluate_squad) (in module neural_compressor.metric.f1) metric_registry() (in module neural_compressor.metric.metric) METRICS (class in neural_compressor.metric.metric) metrics (neural_compressor.metric.metric.METRICS attribute) (neural_compressor.metric.metric.ONNXRTITMetrics attribute) (neural_compressor.metric.metric.ONNXRTQLMetrics attribute) (neural_compressor.metric.metric.PyTorchMetrics attribute) (neural_compressor.metric.metric.TensorflowMetrics attribute) mIOU (class in neural_compressor.metric.metric) mixed_precision_entry() (in module neural_compressor.torch.quantization.algorithm_entry) MixedPrecisionConfig (class in neural_compressor.config) (class in neural_compressor.torch.quantization.config) MNIST (class in neural_compressor.data.datasets.dataset) Mode (class in neural_compressor.common.utils.constants) MODE (class in neural_compressor.utils.utility) Model (class in neural_compressor.model.model) (class in neural_compressor.tensorflow.utils.model) model_forward() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) model_forward_per_sample() (in module neural_compressor.torch.algorithms.smooth_quant.utility) MODEL_LEVEL (neural_compressor.common.tuning_param.ParamLevel attribute) model_type (neural_compressor.model.tensorflow_model.TensorflowQATModel attribute) ModelSize (class in neural_compressor.objective) ModelWiseTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) module neural_compressor neural_compressor.algorithm neural_compressor.algorithm.algorithm neural_compressor.algorithm.fast_bias_correction neural_compressor.algorithm.smooth_quant neural_compressor.algorithm.weight_correction neural_compressor.common neural_compressor.common.base_config neural_compressor.common.base_tuning neural_compressor.common.tuning_param neural_compressor.common.utils neural_compressor.common.utils.constants neural_compressor.common.utils.logger neural_compressor.common.utils.save_load neural_compressor.common.utils.utility neural_compressor.common.version neural_compressor.config neural_compressor.contrib neural_compressor.contrib.strategy neural_compressor.contrib.strategy.tpe neural_compressor.data neural_compressor.data.datasets neural_compressor.data.datasets.bert_dataset neural_compressor.data.datasets.coco_dataset neural_compressor.data.datasets.dataset neural_compressor.data.datasets.dummy_dataset neural_compressor.data.datasets.dummy_dataset_v2 neural_compressor.data.datasets.imagenet_dataset neural_compressor.data.datasets.style_transfer_dataset neural_compressor.data.filters neural_compressor.data.filters.coco_filter neural_compressor.data.filters.filter neural_compressor.data.transforms neural_compressor.data.transforms.imagenet_transform neural_compressor.data.transforms.postprocess neural_compressor.data.transforms.tokenization neural_compressor.data.transforms.transform neural_compressor.metric neural_compressor.metric.bleu neural_compressor.metric.bleu_util neural_compressor.metric.coco_label_map neural_compressor.metric.coco_tools neural_compressor.metric.evaluate_squad neural_compressor.metric.f1 neural_compressor.metric.metric neural_compressor.mix_precision neural_compressor.model neural_compressor.model.base_model neural_compressor.model.keras_model neural_compressor.model.model neural_compressor.model.nets_factory neural_compressor.model.onnx_model neural_compressor.model.tensorflow_model neural_compressor.model.torch_model neural_compressor.objective neural_compressor.profiling neural_compressor.quantization neural_compressor.strategy neural_compressor.strategy.auto neural_compressor.strategy.auto_mixed_precision neural_compressor.strategy.basic neural_compressor.strategy.bayesian neural_compressor.strategy.conservative neural_compressor.strategy.exhaustive neural_compressor.strategy.hawq_v2 neural_compressor.strategy.mse neural_compressor.strategy.mse_v2 neural_compressor.strategy.random neural_compressor.strategy.strategy neural_compressor.strategy.utils neural_compressor.strategy.utils.constant neural_compressor.strategy.utils.tuning_sampler neural_compressor.strategy.utils.tuning_space neural_compressor.strategy.utils.tuning_structs neural_compressor.strategy.utils.utility neural_compressor.tensorflow neural_compressor.tensorflow.algorithms neural_compressor.tensorflow.algorithms.smoother neural_compressor.tensorflow.algorithms.smoother.calibration neural_compressor.tensorflow.algorithms.smoother.core neural_compressor.tensorflow.algorithms.smoother.scaler neural_compressor.tensorflow.algorithms.static_quant neural_compressor.tensorflow.algorithms.static_quant.keras neural_compressor.tensorflow.algorithms.static_quant.tensorflow neural_compressor.tensorflow.keras neural_compressor.tensorflow.keras.layers neural_compressor.tensorflow.keras.layers.conv2d neural_compressor.tensorflow.keras.layers.dense neural_compressor.tensorflow.keras.layers.depthwise_conv2d neural_compressor.tensorflow.keras.layers.layer_initializer neural_compressor.tensorflow.keras.layers.pool2d neural_compressor.tensorflow.keras.layers.separable_conv2d neural_compressor.tensorflow.keras.quantization neural_compressor.tensorflow.keras.quantization.config neural_compressor.tensorflow.quantization neural_compressor.tensorflow.quantization.algorithm_entry neural_compressor.tensorflow.quantization.autotune neural_compressor.tensorflow.quantization.config neural_compressor.tensorflow.quantization.quantize neural_compressor.tensorflow.quantization.utils neural_compressor.tensorflow.quantization.utils.graph_converter neural_compressor.tensorflow.quantization.utils.graph_rewriter neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16 neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8 neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern neural_compressor.tensorflow.quantization.utils.graph_util neural_compressor.tensorflow.quantization.utils.quantize_graph neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_bn neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_concatv2 neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_conv neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_matmul neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_pooling neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_bn neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_concatv2 neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_conv neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_matmul neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling neural_compressor.tensorflow.quantization.utils.quantize_graph_common neural_compressor.tensorflow.quantization.utils.transform_graph neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat neural_compressor.tensorflow.quantization.utils.utility neural_compressor.tensorflow.utils neural_compressor.tensorflow.utils.constants neural_compressor.tensorflow.utils.data neural_compressor.tensorflow.utils.model neural_compressor.tensorflow.utils.model_wrappers neural_compressor.tensorflow.utils.utility neural_compressor.torch neural_compressor.torch.algorithms neural_compressor.torch.algorithms.autoround neural_compressor.torch.algorithms.autoround.autoround neural_compressor.torch.algorithms.base_algorithm neural_compressor.torch.algorithms.layer_wise neural_compressor.torch.algorithms.layer_wise.load neural_compressor.torch.algorithms.layer_wise.modified_pickle neural_compressor.torch.algorithms.layer_wise.utils neural_compressor.torch.algorithms.mixed_precision neural_compressor.torch.algorithms.mixed_precision.half_precision_convert neural_compressor.torch.algorithms.mixed_precision.module_wrappers neural_compressor.torch.algorithms.mx_quant neural_compressor.torch.algorithms.mx_quant.mx neural_compressor.torch.algorithms.mx_quant.utils neural_compressor.torch.algorithms.pt2e_quant neural_compressor.torch.algorithms.pt2e_quant.core neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter neural_compressor.torch.algorithms.pt2e_quant.save_load neural_compressor.torch.algorithms.pt2e_quant.utility neural_compressor.torch.algorithms.qat neural_compressor.torch.algorithms.qat.quant_linear neural_compressor.torch.algorithms.qat.quant_utils neural_compressor.torch.algorithms.qat.tensor_quantizer neural_compressor.torch.algorithms.smooth_quant neural_compressor.torch.algorithms.smooth_quant.save_load neural_compressor.torch.algorithms.smooth_quant.smooth_quant neural_compressor.torch.algorithms.smooth_quant.utility neural_compressor.torch.algorithms.static_quant neural_compressor.torch.algorithms.static_quant.save_load neural_compressor.torch.algorithms.static_quant.static_quant neural_compressor.torch.algorithms.static_quant.utility neural_compressor.torch.algorithms.weight_only neural_compressor.torch.algorithms.weight_only.awq neural_compressor.torch.algorithms.weight_only.gptq neural_compressor.torch.algorithms.weight_only.hqq neural_compressor.torch.algorithms.weight_only.hqq.bitpack neural_compressor.torch.algorithms.weight_only.hqq.config neural_compressor.torch.algorithms.weight_only.hqq.core neural_compressor.torch.algorithms.weight_only.hqq.optimizer neural_compressor.torch.algorithms.weight_only.hqq.qtensor neural_compressor.torch.algorithms.weight_only.hqq.quantizer neural_compressor.torch.algorithms.weight_only.modules neural_compressor.torch.algorithms.weight_only.rtn neural_compressor.torch.algorithms.weight_only.save_load neural_compressor.torch.algorithms.weight_only.teq neural_compressor.torch.algorithms.weight_only.utility neural_compressor.torch.export neural_compressor.torch.export.export_hf neural_compressor.torch.export.pt2e_export neural_compressor.torch.quantization neural_compressor.torch.quantization.algorithm_entry neural_compressor.torch.quantization.autotune neural_compressor.torch.quantization.config neural_compressor.torch.quantization.quantize neural_compressor.torch.quantization.save_load_entry neural_compressor.torch.utils neural_compressor.torch.utils.auto_accelerator neural_compressor.torch.utils.bit_packer neural_compressor.torch.utils.block_wise neural_compressor.torch.utils.constants neural_compressor.torch.utils.environ neural_compressor.torch.utils.utility neural_compressor.training neural_compressor.utils neural_compressor.utils.collect_layer_histogram neural_compressor.utils.constant neural_compressor.utils.create_obj_from_config neural_compressor.utils.export neural_compressor.utils.export.qlinear2qdq neural_compressor.utils.export.tf2onnx neural_compressor.utils.export.torch2onnx neural_compressor.utils.kl_divergence neural_compressor.utils.load_huggingface neural_compressor.utils.logger neural_compressor.utils.options neural_compressor.utils.pytorch neural_compressor.utils.utility neural_compressor.utils.weights_details neural_compressor.version

move_input_device() (in module neural_compressor.torch.utils.utility) move_input_to_device() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) MoveSqueezeAfterReluOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu) MSE (class in neural_compressor.metric.metric) mse (neural_compressor.metric.metric.RMSE attribute) mse_metric_gap() (in module neural_compressor.utils.utility) MSE_V2TuneStrategy (class in neural_compressor.strategy.mse_v2) MSETuneStrategy (class in neural_compressor.strategy.mse) MulLinear (class in neural_compressor.torch.algorithms.weight_only.modules) MultiObjective (class in neural_compressor.objective) mx_quant_entry() (in module neural_compressor.torch.quantization.algorithm_entry) MXLinear (class in neural_compressor.torch.algorithms.mx_quant.mx) MXQuantConfig (class in neural_compressor.torch.quantization.config) MXQuantizer (class in neural_compressor.torch.algorithms.mx_quant.mx)

N

name (neural_compressor.common.base_config.BaseConfig attribute) NASConfig (class in neural_compressor.config) nbits (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute) need_apply() (in module neural_compressor.tensorflow.quantization.quantize) (in module neural_compressor.torch.quantization.quantize) neural_compressor module neural_compressor.algorithm module neural_compressor.algorithm.algorithm module neural_compressor.algorithm.fast_bias_correction module neural_compressor.algorithm.smooth_quant module neural_compressor.algorithm.weight_correction module neural_compressor.common module neural_compressor.common.base_config module neural_compressor.common.base_tuning module neural_compressor.common.tuning_param module neural_compressor.common.utils module neural_compressor.common.utils.constants module neural_compressor.common.utils.logger module neural_compressor.common.utils.save_load module neural_compressor.common.utils.utility module neural_compressor.common.version module neural_compressor.config module neural_compressor.contrib module neural_compressor.contrib.strategy module neural_compressor.contrib.strategy.tpe module neural_compressor.data module neural_compressor.data.datasets module neural_compressor.data.datasets.bert_dataset module neural_compressor.data.datasets.coco_dataset module neural_compressor.data.datasets.dataset module neural_compressor.data.datasets.dummy_dataset module neural_compressor.data.datasets.dummy_dataset_v2 module neural_compressor.data.datasets.imagenet_dataset module neural_compressor.data.datasets.style_transfer_dataset module neural_compressor.data.filters module neural_compressor.data.filters.coco_filter module neural_compressor.data.filters.filter module neural_compressor.data.transforms module neural_compressor.data.transforms.imagenet_transform module neural_compressor.data.transforms.postprocess module neural_compressor.data.transforms.tokenization module neural_compressor.data.transforms.transform module neural_compressor.metric module neural_compressor.metric.bleu module neural_compressor.metric.bleu_util module neural_compressor.metric.coco_label_map module neural_compressor.metric.coco_tools module neural_compressor.metric.evaluate_squad module neural_compressor.metric.f1 module neural_compressor.metric.metric module neural_compressor.mix_precision module neural_compressor.model module neural_compressor.model.base_model module neural_compressor.model.keras_model module neural_compressor.model.model module neural_compressor.model.nets_factory module neural_compressor.model.onnx_model module neural_compressor.model.tensorflow_model module neural_compressor.model.torch_model module neural_compressor.objective module neural_compressor.profiling module neural_compressor.quantization module neural_compressor.strategy module neural_compressor.strategy.auto module neural_compressor.strategy.auto_mixed_precision module neural_compressor.strategy.basic module neural_compressor.strategy.bayesian module neural_compressor.strategy.conservative module neural_compressor.strategy.exhaustive module neural_compressor.strategy.hawq_v2 module neural_compressor.strategy.mse module neural_compressor.strategy.mse_v2 module neural_compressor.strategy.random module neural_compressor.strategy.strategy module neural_compressor.strategy.utils module neural_compressor.strategy.utils.constant module neural_compressor.strategy.utils.tuning_sampler module neural_compressor.strategy.utils.tuning_space module neural_compressor.strategy.utils.tuning_structs module neural_compressor.strategy.utils.utility module neural_compressor.tensorflow module neural_compressor.tensorflow.algorithms module neural_compressor.tensorflow.algorithms.smoother module neural_compressor.tensorflow.algorithms.smoother.calibration module neural_compressor.tensorflow.algorithms.smoother.core module neural_compressor.tensorflow.algorithms.smoother.scaler module neural_compressor.tensorflow.algorithms.static_quant module neural_compressor.tensorflow.algorithms.static_quant.keras module neural_compressor.tensorflow.algorithms.static_quant.tensorflow module neural_compressor.tensorflow.keras module neural_compressor.tensorflow.keras.layers module neural_compressor.tensorflow.keras.layers.conv2d module neural_compressor.tensorflow.keras.layers.dense module neural_compressor.tensorflow.keras.layers.depthwise_conv2d module neural_compressor.tensorflow.keras.layers.layer_initializer module neural_compressor.tensorflow.keras.layers.pool2d module neural_compressor.tensorflow.keras.layers.separable_conv2d module neural_compressor.tensorflow.keras.quantization module neural_compressor.tensorflow.keras.quantization.config module neural_compressor.tensorflow.quantization module neural_compressor.tensorflow.quantization.algorithm_entry module neural_compressor.tensorflow.quantization.autotune module neural_compressor.tensorflow.quantization.config module neural_compressor.tensorflow.quantization.quantize module neural_compressor.tensorflow.quantization.utils module neural_compressor.tensorflow.quantization.utils.graph_converter module neural_compressor.tensorflow.quantization.utils.graph_rewriter module neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16 module neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert module neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input module

neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer module neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8 module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse module neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation module neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq module neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern module neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq module neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern module neural_compressor.tensorflow.quantization.utils.graph_util module neural_compressor.tensorflow.quantization.utils.quantize_graph module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_bn module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_concatv2 module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_conv module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_matmul module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_pooling module neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_bn module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_concatv2 module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_conv module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_matmul module neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling module neural_compressor.tensorflow.quantization.utils.quantize_graph_common module neural_compressor.tensorflow.quantization.utils.transform_graph module neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction module neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base module neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging module neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat module neural_compressor.tensorflow.quantization.utils.utility module neural_compressor.tensorflow.utils module neural_compressor.tensorflow.utils.constants module neural_compressor.tensorflow.utils.data module neural_compressor.tensorflow.utils.model module neural_compressor.tensorflow.utils.model_wrappers module neural_compressor.tensorflow.utils.utility module neural_compressor.torch module neural_compressor.torch.algorithms module neural_compressor.torch.algorithms.autoround module neural_compressor.torch.algorithms.autoround.autoround module neural_compressor.torch.algorithms.base_algorithm module neural_compressor.torch.algorithms.layer_wise module neural_compressor.torch.algorithms.layer_wise.load module neural_compressor.torch.algorithms.layer_wise.modified_pickle module neural_compressor.torch.algorithms.layer_wise.utils module neural_compressor.torch.algorithms.mixed_precision module neural_compressor.torch.algorithms.mixed_precision.half_precision_convert module neural_compressor.torch.algorithms.mixed_precision.module_wrappers module neural_compressor.torch.algorithms.mx_quant module neural_compressor.torch.algorithms.mx_quant.mx module neural_compressor.torch.algorithms.mx_quant.utils module neural_compressor.torch.algorithms.pt2e_quant module neural_compressor.torch.algorithms.pt2e_quant.core module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter module neural_compressor.torch.algorithms.pt2e_quant.save_load module neural_compressor.torch.algorithms.pt2e_quant.utility module neural_compressor.torch.algorithms.qat module neural_compressor.torch.algorithms.qat.quant_linear module neural_compressor.torch.algorithms.qat.quant_utils module neural_compressor.torch.algorithms.qat.tensor_quantizer module neural_compressor.torch.algorithms.smooth_quant module neural_compressor.torch.algorithms.smooth_quant.save_load module neural_compressor.torch.algorithms.smooth_quant.smooth_quant module neural_compressor.torch.algorithms.smooth_quant.utility module neural_compressor.torch.algorithms.static_quant module neural_compressor.torch.algorithms.static_quant.save_load module neural_compressor.torch.algorithms.static_quant.static_quant module neural_compressor.torch.algorithms.static_quant.utility module neural_compressor.torch.algorithms.weight_only module neural_compressor.torch.algorithms.weight_only.awq module neural_compressor.torch.algorithms.weight_only.gptq module neural_compressor.torch.algorithms.weight_only.hqq module neural_compressor.torch.algorithms.weight_only.hqq.bitpack module neural_compressor.torch.algorithms.weight_only.hqq.config module neural_compressor.torch.algorithms.weight_only.hqq.core module neural_compressor.torch.algorithms.weight_only.hqq.optimizer module neural_compressor.torch.algorithms.weight_only.hqq.qtensor module neural_compressor.torch.algorithms.weight_only.hqq.quantizer module neural_compressor.torch.algorithms.weight_only.modules module neural_compressor.torch.algorithms.weight_only.rtn module neural_compressor.torch.algorithms.weight_only.save_load module neural_compressor.torch.algorithms.weight_only.teq module neural_compressor.torch.algorithms.weight_only.utility module neural_compressor.torch.export module neural_compressor.torch.export.export_hf module neural_compressor.torch.export.pt2e_export module neural_compressor.torch.quantization module neural_compressor.torch.quantization.algorithm_entry module neural_compressor.torch.quantization.autotune module neural_compressor.torch.quantization.config module neural_compressor.torch.quantization.quantize module neural_compressor.torch.quantization.save_load_entry module neural_compressor.torch.utils module neural_compressor.torch.utils.auto_accelerator module neural_compressor.torch.utils.bit_packer module neural_compressor.torch.utils.block_wise module neural_compressor.torch.utils.constants module neural_compressor.torch.utils.environ module neural_compressor.torch.utils.utility module neural_compressor.training module neural_compressor.utils module neural_compressor.utils.collect_layer_histogram module neural_compressor.utils.constant module neural_compressor.utils.create_obj_from_config module neural_compressor.utils.export module neural_compressor.utils.export.qlinear2qdq module neural_compressor.utils.export.tf2onnx module neural_compressor.utils.export.torch2onnx module neural_compressor.utils.kl_divergence module neural_compressor.utils.load_huggingface module neural_compressor.utils.logger module neural_compressor.utils.options module neural_compressor.utils.pytorch module neural_compressor.utils.utility module neural_compressor.utils.weights_details module neural_compressor.version module node_from_map() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm) node_name_from_input() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm) nondigit_punct_re (neural_compressor.metric.bleu.UnicodeRegex attribute) normalize_answer() (in module neural_compressor.metric.f1) NormalizeTFTransform (class in neural_compressor.data.transforms.transform) NormalizeTransform (class in neural_compressor.data.transforms.transform) num_correct (neural_compressor.metric.metric.GeneralTopK attribute) (neural_compressor.metric.metric.TensorflowTopK attribute) num_sample (neural_compressor.metric.metric.GeneralTopK attribute) (neural_compressor.metric.metric.TensorflowTopK attribute)

O

Objective (class in neural_compressor.objective) objective_custom_registry() (in module neural_compressor.objective) objective_registry() (in module neural_compressor.objective) ONNX (class in neural_compressor.config) onnx_qlinear_to_qdq() (in module neural_compressor.utils.export.qlinear2qdq) OnnxBilinearImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) ONNXModel (class in neural_compressor.model.onnx_model) ONNXQlinear2QDQConfig (class in neural_compressor.config) ONNXResizeCropImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) onnxrt (class in neural_compressor.utils.options) ONNXRTBertDataset (class in neural_compressor.data.datasets.bert_dataset) ONNXRTCropToBoundingBox (class in neural_compressor.data.transforms.transform) ONNXRTGLUE (class in neural_compressor.metric.metric) ONNXRTImagenetDataset (class in neural_compressor.data.datasets.imagenet_dataset) ONNXRTITDatasets (class in neural_compressor.data.datasets.dataset) ONNXRTITFilters (class in neural_compressor.data.filters.filter) ONNXRTITMetrics (class in neural_compressor.metric.metric)

ONNXRTITTransforms (class in neural_compressor.data.transforms.transform) ONNXRTQLDatasets (class in neural_compressor.data.datasets.dataset) ONNXRTQLFilters (class in neural_compressor.data.filters.filter) ONNXRTQLMetrics (class in neural_compressor.metric.metric) ONNXRTQLTransforms (class in neural_compressor.data.transforms.transform) OP_LEVEL (neural_compressor.common.tuning_param.ParamLevel attribute) OP_TYPE_LEVEL (neural_compressor.common.tuning_param.ParamLevel attribute) OpEntry (class in neural_compressor.utils.utility) OperatorConfig (class in neural_compressor.tensorflow.keras.quantization.config) (class in neural_compressor.torch.quantization.config) optimize_weights_proximal_legacy() (in module neural_compressor.torch.algorithms.weight_only.hqq.optimizer) OptimizedModel (class in neural_compressor.utils.load_huggingface) OptimizeQDQGraph (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq) Options (class in neural_compressor.config) OpTuningConfig (class in neural_compressor.strategy.utils.tuning_structs) OpTypeWiseTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) OpWiseTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) OrderedDefaultDict (class in neural_compressor.strategy.utils.utility)

P

pack_array_with_numba_b2_c16() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b2_c32() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b2_c64() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b2_c8() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b4_c16() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b4_c32() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b4_c64() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b4_c8() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b8_c16() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b8_c32() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b8_c64() (in module neural_compressor.torch.utils.bit_packer) pack_array_with_numba_b8_c8() (in module neural_compressor.torch.utils.bit_packer) Packer (class in neural_compressor.torch.algorithms.weight_only.hqq.bitpack) packing (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute) PaddedCenterCropTransform (class in neural_compressor.data.transforms.transform) ParamLevel (class in neural_compressor.common.tuning_param) params_list (neural_compressor.common.base_config.BaseConfig attribute) parse_cfgs() (in module neural_compressor.torch.algorithms.static_quant.utility) parse_saved_model() (in module neural_compressor.tensorflow.quantization.utils.utility) ParseDecodeBert (class in neural_compressor.data.datasets.bert_dataset) ParseDecodeCoco (class in neural_compressor.data.datasets.coco_dataset) ParseDecodeImagenet (class in neural_compressor.data.transforms.imagenet_transform) ParseDecodeImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) ParseDecodeVocTransform (class in neural_compressor.data.transforms.transform) patch_hqq_moduile() (in module neural_compressor.torch.algorithms.weight_only.hqq.quantizer) pattern_factory() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter) pattern_to_internal() (in module neural_compressor.strategy.utils.tuning_space) pattern_to_path() (in module neural_compressor.strategy.utils.tuning_space) PatternPair (class in neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter) Performance (class in neural_compressor.objective) PickleError PicklingError PostCseOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse) PostHostConstConverter (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter) Postprocess (class in neural_compressor.data.transforms.postprocess) postprocess_model() (in module neural_compressor.torch.utils.utility) PostTrainingQuantConfig (class in neural_compressor.config) pred_list (neural_compressor.metric.metric.Accuracy attribute) (neural_compressor.metric.metric.MAE attribute) (neural_compressor.metric.metric.MSE attribute)

predictions (neural_compressor.metric.bleu.BLEU attribute) PreOptimization (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize) prepare() (in module neural_compressor.torch.quantization.quantize) prepare_compression() (in module neural_compressor.training) prepare_qat() (in module neural_compressor.torch.quantization.quantize) preprocess_quant_config() (in module neural_compressor.torch.quantization.quantize) preprocess_user_cfg() (in module neural_compressor.strategy.utils.utility) print_op_list() (in module neural_compressor.utils.utility) print_table() (in module neural_compressor.utils.utility) ProcessorType (class in neural_compressor.common.utils.utility) pt2e_dynamic_quant_entry() (in module neural_compressor.torch.quantization.algorithm_entry) pt2e_static_quant_entry() (in module neural_compressor.torch.quantization.algorithm_entry) punct_nondigit_re (neural_compressor.metric.bleu.UnicodeRegex attribute) PyTorch (class in neural_compressor.config) PyTorchAlignImageChannel (class in neural_compressor.data.transforms.transform) PyTorchBaseModel (class in neural_compressor.model.torch_model) PytorchBertDataset (class in neural_compressor.data.datasets.bert_dataset) PytorchCIFAR10 (class in neural_compressor.data.datasets.dataset) PytorchCIFAR100 (class in neural_compressor.data.datasets.dataset) PyTorchCropResizeTransform (class in neural_compressor.data.transforms.transform) PyTorchDatasets (class in neural_compressor.data.datasets.dataset) PytorchFashionMNIST (class in neural_compressor.data.datasets.dataset) PyTorchFilters (class in neural_compressor.data.filters.filter) PyTorchFXModel (class in neural_compressor.model.torch_model) PytorchImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) PyTorchLoss (class in neural_compressor.metric.metric) PyTorchMetrics (class in neural_compressor.metric.metric) PytorchMNIST (class in neural_compressor.data.datasets.dataset) PyTorchModel (class in neural_compressor.model.torch_model) PyTorchNormalizeTransform (class in neural_compressor.data.transforms.transform) PytorchTransform (class in neural_compressor.data.transforms.transform) PyTorchTransforms (class in neural_compressor.data.transforms.transform) PyTorchTranspose (class in neural_compressor.data.transforms.transform) PytorchWrapDataset (class in neural_compressor.data.datasets.dataset) PytorchWrapFunction (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.data.transforms.transform)

Q

QAvgPool2D (class in neural_compressor.tensorflow.keras.layers.pool2d) QConv2D (class in neural_compressor.tensorflow.keras.layers.conv2d) QDense (class in neural_compressor.tensorflow.keras.layers.dense) QDepthwiseConv2D (class in neural_compressor.tensorflow.keras.layers.depthwise_conv2d) qdq_quantize() (in module neural_compressor.torch.algorithms.smooth_quant.smooth_quant) qdq_weight_actor() (in module neural_compressor.torch.algorithms.weight_only.utility) qdq_weight_asym() (in module neural_compressor.torch.algorithms.weight_only.utility) qdq_weight_sym() (in module neural_compressor.torch.algorithms.weight_only.utility) QDQLayer (class in neural_compressor.torch.algorithms.layer_wise.utils) (class in neural_compressor.torch.algorithms.weight_only.modules) QMaxPool2D (class in neural_compressor.tensorflow.keras.layers.pool2d) QSeparableConv2D (class in neural_compressor.tensorflow.keras.layers.separable_conv2d) QTensor (class in neural_compressor.torch.algorithms.weight_only.hqq.qtensor) QTensorConfig (class in neural_compressor.torch.algorithms.weight_only.hqq.config) QTensorMetaInfo (class in neural_compressor.torch.algorithms.weight_only.hqq.qtensor) quant_dequant_w_v1() (in module neural_compressor.torch.algorithms.smooth_quant.utility) quant_dequant_x_v1() (in module neural_compressor.torch.algorithms.smooth_quant.utility) quant_mode_from_pattern() (in module neural_compressor.strategy.utils.tuning_space)

quant_tensor() (in module neural_compressor.torch.algorithms.weight_only.utility) quant_weight_w_scale() (in module neural_compressor.torch.algorithms.weight_only.utility) QuantizationAwareTrainingConfig (class in neural_compressor.config) quantize() (in module neural_compressor.torch.quantization.quantize) quantize_4bit() (in module neural_compressor.torch.algorithms.weight_only.utility) quantize_elemwise_op() (in module neural_compressor.torch.algorithms.mx_quant.utils) quantize_model() (in module neural_compressor.tensorflow.quantization.quantize) quantize_model_with_single_config() (in module neural_compressor.tensorflow.quantization.quantize) quantize_mx_op() (in module neural_compressor.torch.algorithms.mx_quant.utils) QuantizedInput (class in neural_compressor.data.transforms.imagenet_transform) QuantizeGraphBase (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base) QuantizeGraphForIntel (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu) QuantizeGraphHelper (class in neural_compressor.tensorflow.quantization.utils.quantize_graph_common) QuantizeNodeBase (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base) Quantizer (class in neural_compressor.torch.algorithms.base_algorithm) (class in neural_compressor.torch.algorithms.weight_only.gptq) QuantLinear (class in neural_compressor.torch.algorithms.qat.quant_linear) QuantOptions (class in neural_compressor.strategy.utils.utility) QuantType (class in neural_compressor.strategy.utils.utility)

R

RandomCropTFTransform (class in neural_compressor.data.transforms.transform) RandomCropTransform (class in neural_compressor.data.transforms.transform) RandomHorizontalFlip (class in neural_compressor.data.transforms.transform) RandomResizedCropPytorchTransform (class in neural_compressor.data.transforms.transform) RandomResizedCropTFTransform (class in neural_compressor.data.transforms.transform) RandomResizedCropTransform (class in neural_compressor.data.transforms.transform) RandomTuneStrategy (class in neural_compressor.strategy.random) RandomVerticalFlip (class in neural_compressor.data.transforms.transform) RAWGPTQuantizer (class in neural_compressor.torch.algorithms.weight_only.gptq) read_graph() (in module neural_compressor.tensorflow.quantization.utils.utility) read_json_file() (in module neural_compressor.torch.utils.utility) read_squad_examples() (in module neural_compressor.data.transforms.transform) reconstruct_saved_model() (in module neural_compressor.tensorflow.quantization.utils.utility) recover() (in module neural_compressor.utils.utility) recover_forward() (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.block_wise) recover_model_from_json() (in module neural_compressor.torch.algorithms.smooth_quant.save_load) (in module neural_compressor.utils.pytorch) register_accelerator() (in module neural_compressor.torch.utils.auto_accelerator) register_algo() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.torch.utils.utility) register_autotune() (in module neural_compressor.torch.algorithms.smooth_quant.utility) register_config() (in module neural_compressor.common.base_config) register_customer_metric() (in module neural_compressor.metric.metric) register_pack_func() (in module neural_compressor.torch.utils.bit_packer) register_supported_configs_for_fwk() (in module neural_compressor.common.base_config)

register_weight_hooks() (in module neural_compressor.torch.algorithms.layer_wise.utils) RemoveTrainingNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes) RenameBatchNormOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm) replace_forward() (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.block_wise) replace_pattern (neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter.PatternPair attribute) replace_with_quant_linear() (in module neural_compressor.torch.algorithms.qat.quant_utils) replacement_fn() (in module neural_compressor.torch.algorithms.weight_only.hqq.quantizer) RerangeQuantizedConcat (class in neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat) RescaleKerasPretrainTransform (class in neural_compressor.data.transforms.transform) RescaleTFTransform (class in neural_compressor.data.transforms.transform) RescaleTransform (class in neural_compressor.data.transforms.transform) reshape_in_channel_to_last() (in module neural_compressor.torch.algorithms.smooth_quant.utility) reshape_scale_as_input() (in module neural_compressor.torch.algorithms.smooth_quant.utility) reshape_scale_as_weight() (in module neural_compressor.torch.algorithms.smooth_quant.utility) ResizePytorchTransform (class in neural_compressor.data.transforms.transform) ResizeTFTransform (class in neural_compressor.data.transforms.transform) ResizeTransform (class in neural_compressor.data.transforms.transform) ResizeWithAspectRatio (class in neural_compressor.data.transforms.imagenet_transform) ResizeWithRatio (class in neural_compressor.data.transforms.transform) reverted_data_type() (in module neural_compressor.strategy.utils.utility) RMSE (class in neural_compressor.metric.metric) ROC (class in neural_compressor.metric.metric) RoundingMode (class in neural_compressor.torch.algorithms.mx_quant.utils) rtn_entry() (in module neural_compressor.torch.quantization.algorithm_entry) RTNConfig (class in neural_compressor.torch.quantization.config) RTNQuantizer (class in neural_compressor.torch.algorithms.weight_only.rtn)

S

sample (neural_compressor.metric.metric.Accuracy attribute) (neural_compressor.metric.metric.Loss attribute) Sampler (class in neural_compressor.common.base_tuning) save() (in module neural_compressor.torch.algorithms.pt2e_quant.save_load) (in module neural_compressor.torch.algorithms.static_quant.save_load) (in module neural_compressor.torch.algorithms.weight_only.save_load) (in module neural_compressor.torch.quantization.save_load_entry) save_config_mapping() (in module neural_compressor.common.utils.save_load) save_for_huggingface_upstream() (in module neural_compressor.utils.load_huggingface) save_layers_in_shards_iteratively() (in module neural_compressor.torch.algorithms.layer_wise.utils) saved_model_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) SaveLoadFormat (class in neural_compressor.torch.utils.constants) ScaleProPagationTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation) search_clip() (in module neural_compressor.torch.algorithms.weight_only.utility) search_pattern (neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter.PatternPair attribute) secure_check_eval_func() (in module neural_compressor.utils.utility) SelfKnowledgeDistillationLossConfig (class in neural_compressor.config) SequentialSampler (class in neural_compressor.common.base_tuning) (class in neural_compressor.tensorflow.utils.data) set_module() (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.utility) set_random_seed() (in module neural_compressor.common.utils.utility) (in module neural_compressor.utils.utility) set_resume_from() (in module neural_compressor.common.utils.utility) (in module neural_compressor.utils.utility) set_tensorboard() (in module neural_compressor.common.utils.utility) (in module neural_compressor.utils.utility) set_workspace() (in module neural_compressor.common.utils.utility) (in module neural_compressor.utils.utility) shape (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute) ShareQDQForItexYPatternOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern) show_memory_info() (in module neural_compressor.utils.utility) simple_inference() (in module neural_compressor.torch.algorithms.static_quant.utility) singleton() (in module neural_compressor.common.utils.utility) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility)

slim_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) smooth_quant_entry() (in module neural_compressor.tensorflow.quantization.algorithm_entry) (in module neural_compressor.torch.quantization.algorithm_entry) SmoothQuant (class in neural_compressor.algorithm.smooth_quant) (class in neural_compressor.tensorflow.algorithms.smoother.core) SmoothQuantCalibration (class in neural_compressor.tensorflow.algorithms.smoother.calibration) SmoothQuantCalibrationLLM (class in neural_compressor.tensorflow.algorithms.smoother.calibration) SmoothQuantConfig (class in neural_compressor.tensorflow.quantization.config) (class in neural_compressor.torch.quantization.config) SmoothQuantQuantizer (class in neural_compressor.torch.algorithms.smooth_quant.smooth_quant) SmoothQuantSampler (class in neural_compressor.strategy.utils.tuning_sampler) SmoothQuantScaler (class in neural_compressor.tensorflow.algorithms.smoother.scaler) SmoothQuantScalerLLM (class in neural_compressor.tensorflow.algorithms.smoother.scaler) SparseDummyDataset (class in neural_compressor.data.datasets.dummy_dataset_v2) SplitSharedInputOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input) SQLinearWrapper (class in neural_compressor.torch.algorithms.smooth_quant.utility) SquadExample (class in neural_compressor.data.transforms.transform) SquadF1 (class in neural_compressor.metric.metric) static_quant_entry() (in module neural_compressor.tensorflow.quantization.algorithm_entry) (in module neural_compressor.torch.quantization.algorithm_entry) static_quant_export() (in module neural_compressor.utils.export.torch2onnx) StaticQuantConfig (class in neural_compressor.tensorflow.keras.quantization.config) (class in neural_compressor.tensorflow.quantization.config) (class in neural_compressor.torch.quantization.config) StaticQuantQuantizer (class in neural_compressor.torch.algorithms.static_quant.static_quant) Statistics (class in neural_compressor.common.utils.utility) (class in neural_compressor.utils.utility) str2array() (in module neural_compressor.utils.utility) strategy_registry() (in module neural_compressor.strategy.strategy) strip_equivalent_nodes() (in module neural_compressor.tensorflow.quantization.utils.utility) strip_unused_nodes() (in module neural_compressor.tensorflow.quantization.utils.utility) StripEquivalentNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes) StripUnusedNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes) StyleTransferDataset (class in neural_compressor.data.datasets.style_transfer_dataset) sum (neural_compressor.metric.metric.Loss attribute) SwitchOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer) symbol_re (neural_compressor.metric.bleu.UnicodeRegex attribute)

T

TargetSpace (class in neural_compressor.strategy.bayesian) TensorFlow (class in neural_compressor.config) Tensorflow (class in neural_compressor.data.datasets.dataset) Tensorflow_ITEXAdaptor (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorFlowAdaptor (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorflowBaseModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowBertDataset (class in neural_compressor.data.datasets.bert_dataset) TensorflowCheckpointModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowCIFAR10 (class in neural_compressor.data.datasets.dataset) TensorflowCIFAR100 (class in neural_compressor.data.datasets.dataset) TensorflowCOCOMAP (class in neural_compressor.metric.metric) TensorFlowConfig (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorflowConfigConverter (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorflowCropToBoundingBox (class in neural_compressor.data.transforms.transform) TensorflowDatasets (class in neural_compressor.data.datasets.dataset) TensorflowFashionMNIST (class in neural_compressor.data.datasets.dataset) TensorflowFilters (class in neural_compressor.data.filters.filter) TensorflowGlobalConfig (class in neural_compressor.tensorflow.utils.model) TensorflowImagenetDataset (class in neural_compressor.data.datasets.imagenet_dataset) TensorflowImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) TensorflowImageRecord (class in neural_compressor.data.datasets.dataset) TensorflowLLMModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowMAP (class in neural_compressor.metric.metric) TensorflowMetrics (class in neural_compressor.metric.metric) TensorflowMNIST (class in neural_compressor.data.datasets.dataset) TensorflowModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowModelZooBertDataset (class in neural_compressor.data.datasets.bert_dataset) TensorflowQATModel (class in neural_compressor.model.tensorflow_model) TensorflowQuery (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorflowRandomHorizontalFlip (class in neural_compressor.data.transforms.transform) TensorflowRandomVerticalFlip (class in neural_compressor.data.transforms.transform) TensorflowResizeCropImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) TensorflowResizeWithRatio (class in neural_compressor.data.transforms.transform) TensorflowSavedModelModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowShiftRescale (class in neural_compressor.data.transforms.imagenet_transform) TensorflowTFRecordDataset (class in neural_compressor.data.datasets.dataset) TensorflowTopK (class in neural_compressor.metric.metric) TensorflowTransform (class in neural_compressor.data.transforms.transform) TensorflowTransforms (class in neural_compressor.data.transforms.transform) TensorflowTranspose (class in neural_compressor.data.transforms.transform) TensorflowTransposeLastChannel (class in neural_compressor.data.transforms.imagenet_transform)

TensorflowVOCMAP (class in neural_compressor.metric.metric) TensorflowVOCRecord (class in neural_compressor.data.datasets.dataset) TensorflowWrapFunction (class in neural_compressor.data.transforms.transform) TensorQuantizer (class in neural_compressor.torch.algorithms.qat.tensor_quantizer) teq_quantize_entry() (in module neural_compressor.torch.quantization.algorithm_entry) TEQConfig (class in neural_compressor.torch.quantization.config) TEQLinearFakeQuant (class in neural_compressor.torch.algorithms.weight_only.modules) TEQuantizer (class in neural_compressor.torch.algorithms.weight_only.teq) TF2ONNXConfig (class in neural_compressor.config) tf_to_fp32_onnx() (in module neural_compressor.utils.export.tf2onnx) tf_to_int8_onnx() (in module neural_compressor.utils.export.tf2onnx) TFModelZooCollectTransform (class in neural_compressor.data.transforms.transform) TFSlimNetsFactory (class in neural_compressor.model.nets_factory) (class in neural_compressor.tensorflow.utils.utility) TFSquadV1ModelZooPostTransform (class in neural_compressor.data.transforms.transform) TFSquadV1PostTransform (class in neural_compressor.data.transforms.transform) time_limit() (in module neural_compressor.utils.utility) to_device() (in module neural_compressor.torch.utils.utility) to_dtype() (in module neural_compressor.torch.utils.utility) ToArray (class in neural_compressor.data.transforms.transform) Torch2ONNXConfig (class in neural_compressor.config) torch_to_fp32_onnx() (in module neural_compressor.utils.export.torch2onnx) torch_to_int8_onnx() (in module neural_compressor.utils.export.torch2onnx) TorchBaseConfig (class in neural_compressor.torch.quantization.config) TorchSmoothQuant (class in neural_compressor.torch.algorithms.smooth_quant.utility) TpeTuneStrategy (class in neural_compressor.contrib.strategy.tpe) trace_gptq_target_blocks() (in module neural_compressor.torch.algorithms.weight_only.gptq) TrainableEquivalentTransformation (class in neural_compressor.torch.algorithms.weight_only.teq) transform_registry() (in module neural_compressor.data.transforms.transform) transformation() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter) TransformerBasedModelBlockPatternDetector (class in neural_compressor.torch.algorithms.static_quant.utility) TRANSFORMS (class in neural_compressor.data.transforms.transform) Transforms (class in neural_compressor.data.transforms.transform) Transpose (class in neural_compressor.data.transforms.transform) try_loading_keras() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) TuneStrategy (class in neural_compressor.strategy.strategy) TuneStrategyMeta (class in neural_compressor.strategy.strategy) TuningConfig (class in neural_compressor.common.base_tuning) TuningCriterion (class in neural_compressor.config) TuningItem (class in neural_compressor.strategy.utils.tuning_space) TuningLogger (class in neural_compressor.common.utils.logger) TuningMonitor (class in neural_compressor.common.base_tuning) TuningOrder (class in neural_compressor.strategy.utils.tuning_sampler) TuningParam (class in neural_compressor.common.tuning_param) TuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) TuningSpace (class in neural_compressor.strategy.utils.tuning_space)

U

UnicodeRegex (class in neural_compressor.metric.bleu) UnpackedWeightOnlyLinearParams (class in neural_compressor.torch.algorithms.weight_only.modules)

UnpicklingError update_module() (in module neural_compressor.torch.algorithms.layer_wise.utils) update_sq_scale() (in module neural_compressor.torch.algorithms.smooth_quant.utility)

V

valid_keras_format() (in module neural_compressor.tensorflow.utils.utility) valid_reshape_inputs() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) validate_and_inference_input_output() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) validate_graph_node() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) validate_modules() (in module neural_compressor.torch.utils.utility) values_from_const() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm)

version1_eq_version2() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_gt_version2() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_gte_version2() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_lt_version2() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_lte_version2() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility)

W

W8A8PT2EQuantizer (class in neural_compressor.torch.algorithms.pt2e_quant.core) warn() (in module neural_compressor.utils.logger) warning() (in module neural_compressor.utils.logger) WeightCorrection (class in neural_compressor.algorithm.weight_correction) WeightOnlyLinear (class in neural_compressor.torch.algorithms.weight_only.modules) WeightOnlyQuantSampler (class in neural_compressor.strategy.utils.tuning_sampler) WeightPruningConfig (class in neural_compressor.config) WeightsDetails (class in neural_compressor.utils.weights_details)

WeightsStatistics (class in neural_compressor.utils.weights_details) whitespace_tokenize() (in module neural_compressor.data.transforms.tokenization) WOQModelLoader (class in neural_compressor.torch.algorithms.weight_only.save_load) WordpieceTokenizer (class in neural_compressor.data.transforms.tokenization) WrapONNXRTMetric (class in neural_compressor.metric.metric) WrapperLayer (class in neural_compressor.torch.algorithms.smooth_quant.utility) WrapPyTorchMetric (class in neural_compressor.metric.metric) write_graph() (in module neural_compressor.tensorflow.quantization.utils.utility) write_json_file() (in module neural_compressor.torch.utils.utility)

X

XPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)