Index

_

_epoch_ran (neural_compressor.compression.callbacks.DistillationCallbacks attribute) (neural_compressor.experimental.distillation.Distillation attribute)

A

Accuracy (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) (class in neural_compressor.objective) AccuracyCriterion (class in neural_compressor.conf.pythonic_config) (class in neural_compressor.config) acq_max() (in module neural_compressor.experimental.strategy.bayesian) (in module neural_compressor.strategy.bayesian) ActAwareWeightQuant (class in neural_compressor.adaptor.torch_utils.awq) ActivationOperator (class in neural_compressor.adaptor.ox_utils.operators.activation) Adaptor (class in neural_compressor.adaptor.adaptor) adaptor_registry() (in module neural_compressor.adaptor.adaptor) add_port_to_name() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) 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) (class in neural_compressor.experimental.data.transforms.transform) alpha (neural_compressor.compression.pruner.regs.GroupLasso attribute) amp_convert() (in module neural_compressor.adaptor.mxnet_utils.util) append_attr() (in module neural_compressor.adaptor.torch_utils.util) apply_awq_clip() (in module neural_compressor.adaptor.ox_utils.weight_only) apply_awq_on_model() (in module neural_compressor.onnxrt.algorithms.weight_only.awq) apply_awq_scale() (in module neural_compressor.adaptor.ox_utils.weight_only) apply_gptq_on_model() (in module neural_compressor.onnxrt.algorithms.weight_only.gptq) apply_inlining() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility)

apply_rtn_on_model() (in module neural_compressor.onnxrt.algorithms.weight_only.rtn) are_shapes_equal() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) ArgMaxOperator (class in neural_compressor.adaptor.ox_utils.operators.argmax) assert_error() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) AttentionOperator (class in neural_compressor.adaptor.ox_utils.operators.attention) attr1 (neural_compressor.template.api_doc_example.ExampleClass attribute) attr2 (neural_compressor.template.api_doc_example.ExampleClass attribute) attr5 (neural_compressor.template.api_doc_example.ExampleClass attribute) attribute1 (in module neural_compressor.template.api_doc_example) attribute_to_kwarg() (in module neural_compressor.adaptor.ox_utils.util) Auto_Accelerator (class in neural_compressor.torch.utils.auto_accelerator) auto_copy() (in module neural_compressor.adaptor.torch_utils.util) autocast (class in neural_compressor.torch.amp.autocast) AutoMixedPrecisionTuneStrategy (class in neural_compressor.experimental.strategy.auto_mixed_precision) (class in neural_compressor.strategy.auto_mixed_precision) autoround_quantize() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.autoround) autotune() (in module neural_compressor.onnxrt.quantization.autotune) (in module neural_compressor.tensorflow.quantization.autotune) (in module neural_compressor.torch.quantization.autotune) AutoTuneStrategy (class in neural_compressor.strategy.auto) awq_quantize() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.awq) (in module neural_compressor.torch.algorithms.weight_only.awq) awq_quantize_entry() (in module neural_compressor.onnxrt.quantization.algorithm_entry) AWQConfig (class in neural_compressor.onnxrt.quantization.config)

B

BaseCallbacks (class in neural_compressor.compression.callbacks) BaseConfig (class in neural_compressor.common.base_config) BaseDataLoader (class in neural_compressor.data.dataloaders.base_dataloader) (class in neural_compressor.experimental.data.dataloaders.base_dataloader) (class in neural_compressor.tensorflow.utils.data) BaseMetric (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) BaseModel (class in neural_compressor.model.base_model) (class in neural_compressor.tensorflow.utils.model_wrappers) BasePattern (class in neural_compressor.compression.pruner.patterns.base) BasePruner (class in neural_compressor.compression.pruner.pruners.base) BasePruning (class in neural_compressor.compression.pruner.pruning) BaseReg (class in neural_compressor.compression.pruner.regs) BaseSearchSpace (class in neural_compressor.compression.hpo.search_space) BaseTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) BasicMagnitudePruner (class in neural_compressor.experimental.pruner_legacy.magnitude) BasicNAS (class in neural_compressor.experimental.nas.basic_nas) BasicPruning (class in neural_compressor.compression.pruner.pruning) BasicTokenizer (class in neural_compressor.data.transforms.tokenization) (class in neural_compressor.experimental.data.transforms.tokenization) BasicTuneStrategy (class in neural_compressor.experimental.strategy.basic) (class in neural_compressor.strategy.basic) BatchNormalizationOperator (class in neural_compressor.adaptor.ox_utils.operators.norm) BatchSampler (class in neural_compressor.data.dataloaders.sampler) (class in neural_compressor.experimental.data.dataloaders.sampler) (class in neural_compressor.tensorflow.utils.data) BayesianOptimization (class in neural_compressor.experimental.strategy.bayesian) (class in neural_compressor.strategy.bayesian) BayesianOptimizationSearcher (class in neural_compressor.compression.hpo.search_algorithms) (class in neural_compressor.experimental.nas.search_algorithms) BayesianTuneStrategy (class in neural_compressor.experimental.strategy.bayesian) (class in neural_compressor.strategy.bayesian) Benchmark (class in neural_compressor.experimental.benchmark)

Benchmark_Conf (class in neural_compressor.conf.config) benchmark_with_raw_cmd() (in module neural_compressor.benchmark) BenchmarkConfig (class in neural_compressor.conf.pythonic_config) (class in neural_compressor.config) best_model (neural_compressor.compression.callbacks.DistillationCallbacks attribute) (neural_compressor.experimental.distillation.Distillation attribute) best_score (neural_compressor.compression.callbacks.DistillationCallbacks attribute) (neural_compressor.experimental.distillation.Distillation attribute) bf16_symbolic_trace() (in module neural_compressor.adaptor.torch_utils.bf16_convert) BF16Convert (class in neural_compressor.adaptor.tf_utils.graph_rewriter.bf16.bf16_convert) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert) BF16ModuleWrapper (class in neural_compressor.adaptor.torch_utils.bf16_convert) BiasCorrection (class in neural_compressor.adaptor.tf_utils.transform_graph.bias_correction) (class in neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction) BilinearImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) BinaryDirect8BitOperator (class in neural_compressor.adaptor.ox_utils.operators.binary_op) BinaryOperator (class in neural_compressor.adaptor.ox_utils.operators.binary_op) BLEU (class in neural_compressor.experimental.metric.bleu) (class in neural_compressor.metric.bleu) bleu_tokenize() (in module neural_compressor.experimental.metric.bleu) (in module neural_compressor.metric.bleu) block_mask module block_size (neural_compressor.compression.pruner.patterns.nxm.KerasPatternNxM attribute) (neural_compressor.compression.pruner.patterns.nxm.PytorchPatternNxM attribute) (neural_compressor.experimental.pytorch_pruner.patterns.PatternNxM attribute) BlockFallbackTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler) BlockMaskCriterion (class in neural_compressor.compression.pruner.criteria) build_slave_faker_model() (in module neural_compressor.strategy.utils.utility) bypass_reshape() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (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

calculate_md5() (in module neural_compressor.data.datasets.dataset) (in module neural_compressor.experimental.data.datasets.dataset) calculate_mse() (in module neural_compressor.utils.utility) calculate_quant_min_max() (in module neural_compressor.adaptor.torch_utils.util) calculate_scale_zp() (in module neural_compressor.adaptor.ox_utils.util) calib_model() (in module neural_compressor.adaptor.mxnet_utils.util) calib_registry() (in module neural_compressor.adaptor.ox_utils.calibrator) CalibCollector (class in neural_compressor.adaptor.mxnet_utils.util) CalibData (class in neural_compressor.adaptor.mxnet_utils.util) calibration() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) CalibrationDataReader (class in neural_compressor.onnxrt.quantization.calibrate) Calibrator (class in neural_compressor.onnxrt.algorithms.smoother.calibrator) CalibratorBase (class in neural_compressor.adaptor.ox_utils.calibrator) call_one() (in module neural_compressor.benchmark) CallBacks (class in neural_compressor.training) CaptureOutputToFile (class in neural_compressor.tensorflow.utils.utility) (class in neural_compressor.utils.utility) cast_tensor() (in module neural_compressor.adaptor.ox_utils.util) CastONNXTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CastPyTorchTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CastTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CenterCropTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CenterCropTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) check_cfg_and_qconfig() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.smooth_quant.utility) (in module neural_compressor.torch.algorithms.static_quant.utility) check_config() (in module neural_compressor.compression.pruner.utils) (in module neural_compressor.experimental.pytorch_pruner.prune_utils) check_dataloader() (in module neural_compressor.data.dataloaders.dataloader) check_integrity() (in module neural_compressor.data.datasets.dataset) (in module neural_compressor.experimental.data.datasets.dataset) check_key_exist() (in module neural_compressor.utils.utility) check_key_validity() (in module neural_compressor.compression.pruner.utils) check_model() (in module neural_compressor.experimental.export.qlinear2qdq) check_model_with_infer_shapes() (in module neural_compressor.onnxrt.utils.utility) check_mx_version() (in module neural_compressor.adaptor.mxnet_utils.util) 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) (class in neural_compressor.experimental.data.datasets.dataset) CIFAR100 (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) ClassifierHeadSearcher (class in neural_compressor.compression.pruner.model_slim.pattern_analyzer) ClassifierHeadSearcherTF (class in neural_compressor.compression.pruner.model_slim.pattern_analyzer) ClassRegister (class in neural_compressor.strategy.utils.utility) COCOEvalWrapper (class in neural_compressor.experimental.metric.coco_tools) (class in neural_compressor.metric.coco_tools) COCOmAPv2 (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) COCONpy (class in neural_compressor.data.datasets.coco_dataset) (class in neural_compressor.experimental.data.datasets.coco_dataset) COCORaw (class in neural_compressor.data.datasets.coco_dataset) (class in neural_compressor.experimental.data.datasets.coco_dataset) COCORecordDataset (class in neural_compressor.data.datasets.coco_dataset) (class in neural_compressor.experimental.data.datasets.coco_dataset) COCOWrapper (class in neural_compressor.experimental.metric.coco_tools) (class in neural_compressor.metric.coco_tools) collate_preds() (in module neural_compressor.adaptor.ox_utils.util) collate_tf_preds() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) collate_torch_preds() (in module neural_compressor.adaptor.torch_utils.util) collect_layer_inputs() (in module neural_compressor.compression.pruner.utils) collect_weight_info() (in module neural_compressor.adaptor.torch_utils.util) CollectorBase (class in neural_compressor.adaptor.mxnet_utils.util) CollectTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) combine_capabilities() (in module neural_compressor.adaptor.mxnet_utils.util) combine_histogram() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) compare_label (neural_compressor.experimental.metric.metric.MAE attribute) (neural_compressor.experimental.metric.metric.MSE attribute) (neural_compressor.metric.metric.MAE attribute) (neural_compressor.metric.metric.MSE attribute) compare_objects() (in module neural_compressor.utils.utility) compare_weights() (in module neural_compressor.adaptor.torch_utils.hawq_metric) Component (class in neural_compressor.experimental.component) ComposableConfig (class in neural_compressor.common.base_config) ComposeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform)

CompressionManager (class in neural_compressor.training) compute_bleu() (in module neural_compressor.experimental.metric.bleu_util) (in module neural_compressor.metric.bleu_util) compute_const_folding_using_tf() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) compute_sparsity() (in module neural_compressor.utils.utility) ConcatOperator (class in neural_compressor.adaptor.ox_utils.operators.concat) Conf (class in neural_compressor.conf.config) conf (neural_compressor.experimental.pruning.Pruning attribute) (neural_compressor.experimental.pruning_v2.Pruning attribute) config (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.compression.pruner.schedulers.PruningScheduler attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) (neural_compressor.experimental.pytorch_pruner.scheduler.Scheduler attribute) config_file_path (neural_compressor.compression.pruner.pruning.BasePruning attribute) (neural_compressor.compression.pruner.pruning.BasicPruning attribute) (neural_compressor.compression.pruner.pruning.RetrainFreePruning attribute) (neural_compressor.experimental.pytorch_pruner.pruning.Pruning attribute) config_instance() (in module neural_compressor.benchmark) config_quantizable_layers() (in module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.optimize_layer) (in module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.optimize_layer) ConservativeTuneStrategy (class in neural_compressor.strategy.conservative) construct_function_from_graph_def() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) ContinuousSearchSpace (class in neural_compressor.compression.hpo.search_space) Convert() (in module neural_compressor.adaptor.torch_utils.bf16_convert) convert() (in module neural_compressor.torch.quantization.quantize) convert_by_vocab() (in module neural_compressor.data.transforms.tokenization) (in module neural_compressor.experimental.data.transforms.tokenization) convert_examples_to_features() (in module neural_compressor.data.datasets.bert_dataset) (in module neural_compressor.data.transforms.transform) (in module neural_compressor.experimental.data.datasets.bert_dataset) (in module neural_compressor.experimental.data.transforms.transform) convert_tensorflow_tensor_to_onnx() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) convert_to_unicode() (in module neural_compressor.data.transforms.tokenization) (in module neural_compressor.experimental.data.transforms.tokenization) ConvertAddToBiasAddOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_add_to_biasadd) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd) ConvertLayoutOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_layout) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout) ConvertLeakyReluOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_leakyrelu) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu) ConvertNanToRandom (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_nan_to_random) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random) ConvertPlaceholderToConst (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_placeholder_to_const) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const) ConvOperator (class in neural_compressor.adaptor.ox_utils.operators.conv) 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_data_example() (in module neural_compressor.adaptor.mxnet_utils.util) 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_onnx_config() (in module neural_compressor.profiling.profiler.onnxrt_profiler.utils) create_search_space_pool() (in module neural_compressor.experimental.nas.nas_utils) create_tf_config() (in module neural_compressor.profiling.profiler.tensorflow_profiler.utils) create_train_func() (in module neural_compressor.utils.create_obj_from_config) criterion (neural_compressor.compression.pruner.pruners.basic.KerasBasicPruner attribute) (neural_compressor.compression.pruner.pruners.basic.PytorchBasicPruner attribute) (neural_compressor.compression.pruner.pruners.block_mask.PytorchBlockMaskPruner attribute) (neural_compressor.compression.pruner.pruners.retrain_free.PytorchRetrainFreePruner attribute) (neural_compressor.compression.pruner.pruners.sparse_gpt.SparseGPTPruner attribute) criterion_registry() (in module neural_compressor.compression.distillation.criterions) (in module neural_compressor.experimental.common.criterion) Criterions (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) CropResizeTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CropResizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CropToBoundingBox (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) CUDA_Accelerator (class in neural_compressor.torch.utils.auto_accelerator) current_pattern (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) current_sparsity_ratio (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute)

D

DataIterLoader (class in neural_compressor.adaptor.mxnet_utils.util) DataLoader (class in neural_compressor.data.dataloaders.dataloader) (class in neural_compressor.experimental.common.dataloader) DataLoaderWrap (class in neural_compressor.adaptor.mxnet_utils.util) Dataset (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) dataset (neural_compressor.experimental.metric.coco_tools.COCOWrapper attribute) (neural_compressor.metric.coco_tools.COCOWrapper attribute) dataset_registry() (in module neural_compressor.data.datasets.dataset) (in module neural_compressor.experimental.data.datasets.dataset) Datasets (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) debug() (in module neural_compressor.utils.logger) deep_get() (in module neural_compressor.conf.dotdict) (in module neural_compressor.tensorflow.utils.utility) deep_set() (in module neural_compressor.conf.dotdict) default_collate() (in module neural_compressor.data.dataloaders.default_dataloader) (in module neural_compressor.experimental.data.dataloaders.default_dataloader) (in module neural_compressor.tensorflow.utils.data) DefaultConf (class in neural_compressor.conf.config) DefaultDataLoader (class in neural_compressor.data.dataloaders.default_dataloader) (class in neural_compressor.experimental.data.dataloaders.default_dataloader) delete_assign() (in module neural_compressor.profiling.profiler.tensorflow_profiler.utils) Dequantize() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) dequantize_data() (in module neural_compressor.adaptor.ox_utils.util) dequantize_data_with_scale_zero() (in module neural_compressor.adaptor.ox_utils.util) dequantize_weight() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) DequantizeCastOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.bf16.dequantize_cast_optimizer) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer) detection_type (neural_compressor.experimental.metric.coco_tools.COCOWrapper attribute) (neural_compressor.metric.coco_tools.COCOWrapper attribute) device (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcherTF attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.JitBasicSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.SelfMHASearcher attribute) (neural_compressor.compression.pruner.model_slim.weight_slim.LinearCompression attribute)

DilatedContraction (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.dilated_contraction) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction) Direct8BitOperator (class in neural_compressor.adaptor.ox_utils.operators.direct_q8) disable_random() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.utils.utility) DiscreteSearchSpace (class in neural_compressor.compression.hpo.search_space) Distillation (class in neural_compressor.experimental.distillation) Distillation_Conf (class in neural_compressor.conf.config) DistillationCallbacks (class in neural_compressor.compression.callbacks) DistillationConfig (class in neural_compressor.conf.pythonic_config) (class in neural_compressor.config) distribute_calib_tensors() (in module neural_compressor.adaptor.mxnet_utils.util) DotDict (class in neural_compressor.conf.dotdict) (class in neural_compressor.config) (class in neural_compressor.utils.utility) dowload_hf_model() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) download_url() (in module neural_compressor.data.datasets.dataset) (in module neural_compressor.experimental.data.datasets.dataset) DSnoT() (in module neural_compressor.compression.pruner.dsnot) dtype_to_name() (in module neural_compressor.adaptor.ox_utils.util) DummyDataset (class in neural_compressor.data.datasets.dummy_dataset) (class in neural_compressor.data.datasets.dummy_dataset_v2) (class in neural_compressor.experimental.data.datasets.dummy_dataset) (class in neural_compressor.experimental.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.tensorflow.utils.utility) (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.torch.algorithms.weight_only.hqq.utility) (in module neural_compressor.utils.utility) dump_model_op_stats() (in module neural_compressor.torch.algorithms.static_quant.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.experimental.export.torch2onnx) DyNAS (class in neural_compressor.experimental.nas.dynas)

E

EmbedLayerNormalizationOperator (class in neural_compressor.adaptor.ox_utils.operators.embed_layernorm) end_step (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) ensure_list() (in module neural_compressor.adaptor.mxnet_utils.util) 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) eval_frequency (neural_compressor.compression.callbacks.DistillationCallbacks attribute) (neural_compressor.experimental.distillation.Distillation attribute) evaluate() (in module neural_compressor.experimental.metric.evaluate_squad) (in module neural_compressor.experimental.metric.f1) (in module neural_compressor.metric.evaluate_squad) (in module neural_compressor.metric.f1) Evaluator (class in neural_compressor.common.base_tuning)

exact_match_score() (in module neural_compressor.experimental.metric.evaluate_squad) (in module neural_compressor.metric.evaluate_squad) ExampleClass (class in neural_compressor.template.api_doc_example) ExhaustiveTuneStrategy (class in neural_compressor.experimental.strategy.exhaustive) (class in neural_compressor.strategy.exhaustive) ExpandDimsOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.expanddims_optimizer) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer) export_compressed_model() (in module neural_compressor.utils.load_huggingface) ExportConfig (class in neural_compressor.config) ExportSingleImageDetectionBoxesToCoco() (in module neural_compressor.experimental.metric.coco_tools) (in module neural_compressor.metric.coco_tools) ExportSingleImageDetectionMasksToCoco() (in module neural_compressor.experimental.metric.coco_tools) (in module neural_compressor.metric.coco_tools) ExportSingleImageGroundtruthToCoco() (in module neural_compressor.experimental.metric.coco_tools) (in module neural_compressor.metric.coco_tools) extract_data_type() (in module neural_compressor.experimental.strategy.utils.utility) (in module neural_compressor.strategy.utils.utility)

F

F1 (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) f1_score() (in module neural_compressor.experimental.metric.evaluate_squad) (in module neural_compressor.experimental.metric.f1) (in module neural_compressor.metric.evaluate_squad) (in module neural_compressor.metric.f1) FakeAffineTensorQuantFunction (class in neural_compressor.adaptor.torch_utils.model_wrapper) (class in neural_compressor.torch.algorithms.weight_only.modules) FakeQuantize (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.fake_quantize) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize) FakeQuantizeBase (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.fake_quantize) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize) FallbackTuningSampler (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) FashionMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.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.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) (in module neural_compressor.torch.utils.utility) Fetcher (class in neural_compressor.data.dataloaders.fetcher) (class in neural_compressor.experimental.data.dataloaders.fetcher) FetchWeightFromReshapeOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fetch_weight_from_reshape) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape) Filter (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) filter_registry() (in module neural_compressor.data.filters.filter) (in module neural_compressor.experimental.data.filters.filter) FILTERS (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) find_by_name() (in module neural_compressor.adaptor.ox_utils.util) (in module neural_compressor.onnxrt.utils.utility) find_layers() (in module neural_compressor.adaptor.torch_utils.gptq) (in module neural_compressor.compression.pruner.wanda.utils) (in module neural_compressor.torch.algorithms.weight_only.gptq) find_layers_name() (in module neural_compressor.adaptor.torch_utils.gptq) (in module neural_compressor.torch.algorithms.weight_only.gptq) find_opset() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) find_pareto_front() (in module neural_compressor.experimental.nas.nas_utils) fit() (in module neural_compressor.benchmark) (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.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) flatten_static_graph (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcherTF attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.JitBasicSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.SelfMHASearcher attribute) Float16ActivationOperator (class in neural_compressor.adaptor.ox_utils.operators.activation) Float16BinaryOperator (class in neural_compressor.adaptor.ox_utils.operators.binary_op) float_to_bfloat16() (in module neural_compressor.adaptor.ox_utils.util) float_to_float16() (in module neural_compressor.adaptor.ox_utils.util) FoldBatchNormNodesOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_batch_norm) (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.adaptor.torch_utils.util) framework_datasets (in module neural_compressor.data.datasets.dataset) (in module neural_compressor.experimental.data.datasets.dataset) FreezeFakeQuantOpOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_fake_quant) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant) FreezeValueTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_value) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value) FreezeValueWithoutCalibTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_value_without_calib) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value_without_calib)

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) (class in neural_compressor.experimental.data.transforms.tokenization) function1() (in module neural_compressor.template.api_doc_example) function2() (in module neural_compressor.template.api_doc_example) function3() (in module neural_compressor.template.api_doc_example) fuse() (in module neural_compressor.adaptor.mxnet_utils.util) FuseBiasAddAndAddOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_biasadd_add) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add) FuseColumnWiseMulOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_column_wise_mul) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul) FuseConvRedundantDequantizeTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_conv_redundant_dequantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize) FuseConvRequantizeTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_conv_requantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize) FuseConvWithMathOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_conv_with_math) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math) FuseDecomposedBNOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn) FuseDecomposedINOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in) FusedMatMulOperator (class in neural_compressor.adaptor.ox_utils.operators.matmul) FuseGeluOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_gelu) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu) FuseLayerNormOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_layer_norm) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm) FuseMatMulRedundantDequantizeTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize) FuseMatMulRequantizeDequantizeNewAPITransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_requantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseMatMulRequantizeDequantizeTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_requantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseMatMulRequantizeNewAPITransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_requantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseMatMulRequantizeTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_requantize) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize) FuseNodeStartWithConcatV2 (class in neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_concatv2) (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_concatv2) (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.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_conv) (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_conv) (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.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_deconv) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv) FuseNodeStartWithFusedBatchNormV3 (class in neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_bn) (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_bn) (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.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_in) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in) FuseNodeStartWithMatmul (class in neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_matmul) (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_matmul) (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.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_pooling) (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_pooling) (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.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_conv) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv) FusePadWithFP32Conv2DOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_fp32_conv) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv) FuseTransposeReshapeOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_reshape_transpose) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose)

G

GatherOperator (class in neural_compressor.adaptor.ox_utils.operators.gather) GemmOperator (class in neural_compressor.adaptor.ox_utils.operators.gemm) gen_bar_updater() (in module neural_compressor.data.datasets.dataset) (in module neural_compressor.experimental.data.datasets.dataset) GeneralTopK (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) generate_activation_observer() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) generate_feed_dict() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) generate_ffn2_pruning_config() (in module neural_compressor.compression.pruner.model_slim.auto_slim) generate_mha_pruning_config() (in module neural_compressor.compression.pruner.model_slim.auto_slim) generate_prefix() (in module neural_compressor.benchmark) generate_pruner_config() (in module neural_compressor.compression.pruner.utils) GenerateGraphWithQDQPattern (class in neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.insert_qdq_pattern) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern) generator1() (in module neural_compressor.template.api_doc_example) get_absorb_layers() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) get_activation() (in module neural_compressor.compression.distillation.utility) (in module neural_compressor.experimental.common.torch_utils) get_adaptor_name() (in module neural_compressor.experimental.strategy.utils.utility) (in module neural_compressor.strategy.utils.utility) get_algorithm() (in module neural_compressor.utils.create_obj_from_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) get_architecture() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) get_attributes() (in module neural_compressor.compression.pruner.model_slim.pattern_analyzer) get_autoround_default_run_fn() (in module neural_compressor.torch.algorithms.weight_only.autoround) get_blob_size() (in module neural_compressor.adaptor.ox_utils.weight_only) get_block_prefix() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) get_bounded_threads() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) get_children() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) get_common_module() (in module neural_compressor.compression.pruner.model_slim.pattern_analyzer) get_const_dim_count() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (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_core_ids() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) get_criterion() (in module neural_compressor.compression.pruner.criteria) get_default_awq_config() (in module neural_compressor.onnxrt.quantization.config) get_default_gptq_config() (in module neural_compressor.onnxrt.quantization.config) (in module neural_compressor.torch.quantization.config) get_default_hqq_config() (in module neural_compressor.torch.quantization.config) get_default_rtn_config() (in module neural_compressor.onnxrt.quantization.config) (in module neural_compressor.torch.quantization.config) get_default_sq_config() (in module neural_compressor.onnxrt.quantization.config) (in module neural_compressor.tensorflow.quantization.config) get_default_static_quant_config() (in module neural_compressor.tensorflow.keras.quantization.config) (in module neural_compressor.tensorflow.quantization.config) get_depth() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) get_dict_at_depth() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) get_element_under_depth() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) get_embedding_contiguous() (in module neural_compressor.adaptor.torch_utils.util) get_estimator_graph() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) get_example_input() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) get_fallback_order() (in module neural_compressor.adaptor.torch_utils.util) get_final_text() (in module neural_compressor.data.transforms.transform) (in module neural_compressor.experimental.data.transforms.transform) get_framework_name() (in module neural_compressor.adaptor.mxnet_utils.util) get_func_from_config() (in module neural_compressor.utils.create_obj_from_config) get_graph_def() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) get_hidden_states() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) get_index_from_strided_slice_of_shape() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) get_input_output_node_names() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) get_layers() (in module neural_compressor.compression.pruner.utils) get_max_supported_opset_version() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema) get_metrics() (in module neural_compressor.utils.create_obj_from_config) get_model_fwk_name() (in module neural_compressor.model.model) get_model_input_shape() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) get_model_path() (in module neural_compressor.utils.neural_insights_utils) get_model_type() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) get_module() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.adaptor.torch_utils.waq.utils) (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) get_module_input_output() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.weight_only.utility) get_mse_order_per_fp32() (in module neural_compressor.adaptor.torch_utils.util) get_mse_order_per_int8() (in module neural_compressor.adaptor.torch_utils.util) get_named_children() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) get_node_mapping() (in module neural_compressor.experimental.export.torch2onnx) get_node_original_name() (in module neural_compressor.adaptor.ox_utils.util)

get_number_of_sockets() (in module neural_compressor.utils.utility) get_op_list() (in module neural_compressor.utils.utility) get_op_type_by_name() (in module neural_compressor.adaptor.torch_utils.util) get_ops_recursively() (in module neural_compressor.adaptor.pytorch) get_pattern() (in module neural_compressor.compression.pruner.patterns) (in module neural_compressor.experimental.pytorch_pruner.patterns) get_physical_ids() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) get_postprocess() (in module neural_compressor.utils.create_obj_from_config) get_preprocess() (in module neural_compressor.utils.create_obj_from_config) get_pruner() (in module neural_compressor.compression.pruner.pruners) (in module neural_compressor.experimental.pytorch_pruner.pruner) get_qrange_for_qType() (in module neural_compressor.onnxrt.utils.utility) get_quant_dequant_output() (in module neural_compressor.adaptor.ox_utils.smooth_quant) get_quantizable_onnx_ops() (in module neural_compressor.experimental.export.torch2onnx) get_quantizable_ops_from_cfgs() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) get_quantizable_ops_recursively() (in module neural_compressor.torch.algorithms.static_quant.utility) get_reg() (in module neural_compressor.compression.pruner.regs) get_reg_type() (in module neural_compressor.compression.pruner.regs) get_scheduler() (in module neural_compressor.compression.pruner.schedulers) (in module neural_compressor.experimental.pytorch_pruner.scheduler) get_schema() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema) get_size() (in module neural_compressor.utils.utility) get_sparsity_ratio() (in module neural_compressor.compression.pruner.utils) get_sparsity_ratio_tf() (in module neural_compressor.compression.pruner.utils) get_subgraphs_from_onnx() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) get_super_module_by_name() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) get_tensor_by_name() (in module neural_compressor.adaptor.tf_utils.util) (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_tensor_val_from_graph_node() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) get_tensorflow_node_attr() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) get_tensorflow_node_shape_attr() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) get_tensorflow_tensor_data() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) get_tensorflow_tensor_shape() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) get_tensors_info() (in module neural_compressor.utils.utility) get_tf_criterion() (in module neural_compressor.compression.pruner.tf_criteria) get_threads() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) get_threads_per_core() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) get_torch_version() (in module neural_compressor.adaptor.torch_utils.util) get_torchvision_map() (in module neural_compressor.data.transforms.transform) (in module neural_compressor.experimental.data.transforms.transform) get_tuning_history() (in module neural_compressor.utils.utility) get_weight_from_input_tensor() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) get_weight_scale() (in module neural_compressor.adaptor.ox_utils.weight_only) get_weights_details() (in module neural_compressor.utils.utility) GLOBAL_STATE (class in neural_compressor.utils.utility) global_step (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) GlobalAveragePoolOperator (class in neural_compressor.adaptor.ox_utils.operators.gavgpool) GPTQ (class in neural_compressor.adaptor.torch_utils.gptq) (class in neural_compressor.torch.algorithms.weight_only.gptq) gptq() (in module neural_compressor.adaptor.ox_utils.weight_only) gptq_quantize() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.gptq) (in module neural_compressor.torch.algorithms.weight_only.gptq) gptq_quantize_entry() (in module neural_compressor.onnxrt.quantization.algorithm_entry) GPTQConfig (class in neural_compressor.onnxrt.quantization.config) (class in neural_compressor.torch.quantization.config) GPTQuantizer (class in neural_compressor.adaptor.torch_utils.gptq) (class in neural_compressor.torch.algorithms.weight_only.gptq) GradientCriterion (class in neural_compressor.compression.pruner.criteria) GradientSensitivityPruner (class in neural_compressor.experimental.pruner_legacy.gradient_sensitivity) graph_def_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) Graph_Optimization (class in neural_compressor.experimental.graph_optimization) Graph_Optimization_Conf (class in neural_compressor.conf.config) graph_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) GraphAnalyzer (class in neural_compressor.adaptor.tf_utils.graph_util) (class in neural_compressor.tensorflow.quantization.utils.graph_util) GraphConverter (class in neural_compressor.adaptor.tf_utils.graph_converter) (class in neural_compressor.tensorflow.quantization.utils.graph_converter) GraphConverterWithoutCalib (class in neural_compressor.adaptor.tf_utils.graph_converter_without_calib) (class in neural_compressor.tensorflow.quantization.utils.graph_converter_without_calib) GraphCseOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.graph_cse_optimizer) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer) GraphFoldConstantOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_constant) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant) GraphRewriterBase (class in neural_compressor.adaptor.tf_utils.graph_rewriter.graph_base) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base) GraphRewriterHelper (class in neural_compressor.adaptor.tf_utils.graph_util) (class in neural_compressor.tensorflow.quantization.utils.graph_util) GraphTransformBase (class in neural_compressor.adaptor.tf_utils.transform_graph.graph_transform_base) (class in neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base) GrapplerOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.grappler_pass) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass) GridSearcher (class in neural_compressor.compression.hpo.search_algorithms) (class in neural_compressor.experimental.nas.search_algorithms) GroupLasso (class in neural_compressor.compression.pruner.regs) GroupLassoPruner (class in neural_compressor.experimental.pruner_legacy.group_lasso)

H

hawq_top() (in module neural_compressor.adaptor.torch_utils.hawq_metric) HAWQ_V2TuneStrategy (class in neural_compressor.strategy.hawq_v2) head_masks (neural_compressor.compression.pruner.pruners.mha.PythonMultiheadAttentionPruner attribute) HessianTrace (class in neural_compressor.adaptor.torch_utils.hawq_metric)

HistogramCollector (class in neural_compressor.adaptor.ox_utils.calibrator) HPOConfig (class in neural_compressor.config) HPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator) HQQConfig (class in neural_compressor.torch.quantization.config) HQQModuleConfig (class in neural_compressor.torch.algorithms.weight_only.hqq.config)

I

ImageFolder (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) ImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) (class in neural_compressor.experimental.data.datasets.imagenet_dataset) IndexFetcher (class in neural_compressor.data.dataloaders.fetcher) (class in neural_compressor.experimental.data.dataloaders.fetcher) (class in neural_compressor.tensorflow.utils.data) infer_onnx_shape_dtype() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) infer_shapes() (in module neural_compressor.adaptor.ox_utils.util) info() (in module neural_compressor.utils.logger) init_quantize_config() (in module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_helper) (in module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_helper) initial_tuning_cfg_with_quant_mode() (in module neural_compressor.experimental.strategy.utils.tuning_space) (in module neural_compressor.strategy.utils.tuning_space) initialize_name_counter() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) InjectDummyBiasAddOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.dummy_biasadd) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd) input2tuple() (in module neural_compressor.adaptor.torch_utils.util) InputFeatures (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.datasets.bert_dataset) (class in neural_compressor.experimental.data.transforms.transform) InsertLogging (class in neural_compressor.adaptor.tf_utils.transform_graph.insert_logging) (class in neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging) InsertPrintMinMaxNode (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.insert_print_node) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node) int8_node_name_reverse() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) IntermediateLayersKnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) IntermediateLayersKnowledgeDistillationLossConfig (class in neural_compressor.config) invalid_layers (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute)

ipex_mixed_precision() (in module neural_compressor.adaptor.torch_utils.mixed_precision) IPEXModel (class in neural_compressor.model.torch_model) is_B_transposed() (in module neural_compressor.adaptor.ox_utils.util) (in module neural_compressor.onnxrt.utils.utility) is_ckpt_format() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) is_fused_module() (in module neural_compressor.adaptor.torch_utils.util) is_global (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) (neural_compressor.experimental.pytorch_pruner.patterns.Pattern attribute) is_int8_model() (in module neural_compressor.utils.pytorch) is_leaf() (in module neural_compressor.adaptor.torch_utils.gptq) (in module neural_compressor.torch.algorithms.weight_only.gptq) is_list_or_tuple() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) is_model_quantized() (in module neural_compressor.adaptor.mxnet_utils.util) is_onnx_domain() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) is_saved_model_format() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) isiterable() (in module neural_compressor.adaptor.mxnet_utils.util) IterableDataset (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) IterableFetcher (class in neural_compressor.data.dataloaders.fetcher) (class in neural_compressor.experimental.data.dataloaders.fetcher) (class in neural_compressor.tensorflow.utils.data) IterableSampler (class in neural_compressor.data.dataloaders.sampler) (class in neural_compressor.experimental.data.dataloaders.sampler) (class in neural_compressor.tensorflow.utils.data) IterativeScheduler (class in neural_compressor.compression.pruner.schedulers) (class in neural_compressor.experimental.pytorch_pruner.scheduler) iterator_sess_run() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) itex_installed() (in module neural_compressor.tensorflow.utils.utility)

J

JitBasicSearcher (class in neural_compressor.compression.pruner.model_slim.pattern_analyzer)

K

k (neural_compressor.experimental.metric.metric.GeneralTopK attribute) (neural_compressor.experimental.metric.metric.TensorflowTopK attribute) (neural_compressor.metric.metric.GeneralTopK attribute) (neural_compressor.metric.metric.TensorflowTopK attribute) keep_mask_layers (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern 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.adaptor.keras) (class in neural_compressor.tensorflow.algorithms.static_quant.keras) KerasBasePattern (class in neural_compressor.compression.pruner.patterns.base) KerasBasePruner (class in neural_compressor.compression.pruner.pruners.base)

KerasBasicPruner (class in neural_compressor.compression.pruner.pruners.basic) 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) KerasPatternNxM (class in neural_compressor.compression.pruner.patterns.nxm) 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) KLCalibrator (class in neural_compressor.adaptor.ox_utils.calibrator) KnowledgeDistillationFramework (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) KnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) KnowledgeDistillationLossConfig (class in neural_compressor.conf.pythonic_config) (class in neural_compressor.config)

L

label_list (neural_compressor.experimental.metric.metric.Accuracy attribute) (neural_compressor.experimental.metric.metric.MAE attribute) (neural_compressor.experimental.metric.metric.MSE attribute) (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) (class in neural_compressor.experimental.data.filters.coco_filter) LabelBalanceCOCORecordFilter (class in neural_compressor.data.filters.coco_filter) (class in neural_compressor.experimental.data.filters.coco_filter) labels (neural_compressor.experimental.metric.bleu.BLEU attribute) (neural_compressor.metric.bleu.BLEU attribute) LabelShift (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) layer_1 (neural_compressor.compression.pruner.model_slim.weight_slim.LinearCompression attribute) layer_2 (neural_compressor.compression.pruner.model_slim.weight_slim.LinearCompression attribute) layer_wise_quant() (in module neural_compressor.onnxrt.algorithms.layer_wise.core) LayerHistogramCollector (class in neural_compressor.utils.collect_layer_histogram) LayerWiseQuant (class in neural_compressor.adaptor.torch_utils.layer_wise_quant.quantize) LazyImport (class in neural_compressor.common.utils.utility) (class in neural_compressor.tensorflow.utils.utility) (class in neural_compressor.utils.utility) Linear2LinearSearcher (class in neural_compressor.compression.pruner.model_slim.pattern_analyzer) linear_layers (neural_compressor.compression.pruner.pruners.mha.PythonMultiheadAttentionPruner attribute) linear_patterns (neural_compressor.compression.pruner.model_slim.weight_slim.LinearCompressionIterator attribute) LinearCompression (class in neural_compressor.compression.pruner.model_slim.weight_slim) LinearCompressionIterator (class in neural_compressor.compression.pruner.model_slim.weight_slim) load() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.torch_load) (in module neural_compressor.torch.algorithms.layer_wise.load) (in module neural_compressor.utils.pytorch)

load_and_cache_examples() (in module neural_compressor.data.datasets.bert_dataset) (in module neural_compressor.experimental.data.datasets.bert_dataset) load_config_mapping() (in module neural_compressor.common.utils.save_load) load_data_from_pkl() (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) load_empty_model() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) load_layer_wise_quantized_model() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) load_saved_model() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) load_tensor() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) load_tensor_from_shard() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) load_vocab() (in module neural_compressor.data.transforms.tokenization) (in module neural_compressor.experimental.data.transforms.tokenization) load_weight_only() (in module neural_compressor.utils.pytorch) log() (in module neural_compressor.utils.logger) log_quantizable_layers_per_transformer() (in module neural_compressor.adaptor.torch_utils.gptq) (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.experimental.metric.metric) (class in neural_compressor.metric.metric) LowerBitsSampler (class in neural_compressor.strategy.utils.tuning_sampler) LSTMOperator (class in neural_compressor.adaptor.ox_utils.operators.lstm)

M

M (neural_compressor.compression.pruner.patterns.mha.PatternMHA attribute) (neural_compressor.compression.pruner.patterns.ninm.PytorchPatternNInM attribute) (neural_compressor.experimental.pytorch_pruner.patterns.PatternNInM attribute) MAE (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) MagnitudeCriterion (class in neural_compressor.compression.pruner.criteria) (class in neural_compressor.compression.pruner.tf_criteria) MagnitudePruner (class in neural_compressor.experimental.pytorch_pruner.pruner) make_dquant_node() (in module neural_compressor.adaptor.ox_utils.util) make_matmul_weight_only_node() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.utility) make_module() (in module neural_compressor.adaptor.mxnet_utils.util) make_nc_model() (in module neural_compressor.adaptor.mxnet_utils.util) make_onnx_inputs_outputs() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) make_onnx_shape() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) make_quant_node() (in module neural_compressor.adaptor.ox_utils.util) make_sub_graph() (in module neural_compressor.adaptor.ox_utils.smooth_quant) make_symbol_block() (in module neural_compressor.adaptor.mxnet_utils.util) map_numpy_to_onnx_dtype() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) map_onnx_to_numpy_type() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) map_tensorflow_dtype() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) masks (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) match_datatype_pattern() (in module neural_compressor.adaptor.torch_utils.util) MatMulOperator (class in neural_compressor.adaptor.ox_utils.operators.matmul) max_sparsity_ratio_per_layer (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) max_sparsity_ratio_per_op (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) MaxPoolOperator (class in neural_compressor.adaptor.ox_utils.operators.maxpool) MergeDuplicatedQDQOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.merge_duplicated_qdq) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq) MetaInfoChangingMemOpOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.meta_op_optimizer) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer) Metric (class in neural_compressor.experimental.common.metric) (class in neural_compressor.metric.metric) metric_max_over_ground_truths() (in module neural_compressor.experimental.metric.evaluate_squad) (in module neural_compressor.experimental.metric.f1) (in module neural_compressor.metric.evaluate_squad) (in module neural_compressor.metric.f1) metric_registry() (in module neural_compressor.experimental.metric.metric) (in module neural_compressor.metric.metric) METRICS (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) metrics (neural_compressor.experimental.metric.metric.METRICS attribute) (neural_compressor.experimental.metric.metric.MXNetMetrics attribute) (neural_compressor.experimental.metric.metric.ONNXRTITMetrics attribute) (neural_compressor.experimental.metric.metric.ONNXRTQLMetrics attribute) (neural_compressor.experimental.metric.metric.PyTorchMetrics attribute) (neural_compressor.experimental.metric.metric.TensorflowMetrics attribute) (neural_compressor.metric.metric.METRICS attribute) (neural_compressor.metric.metric.MXNetMetrics 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) mha_compressions (neural_compressor.compression.pruner.pruners.mha.PythonMultiheadAttentionPruner attribute) mha_scores (neural_compressor.compression.pruner.pruners.mha.PythonMultiheadAttentionPruner attribute) min_sparsity_ratio_per_op (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) MinMaxCalibrator (class in neural_compressor.adaptor.ox_utils.calibrator) mIOU (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) MixedPrecision (class in neural_compressor.experimental.mixed_precision) MixedPrecision_Conf (class in neural_compressor.conf.config) MixedPrecisionConfig (class in neural_compressor.config) MNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) Mode (class in neural_compressor.torch.utils.utility) MODE (class in neural_compressor.utils.utility) Model (class in neural_compressor.experimental.common.model) (class in neural_compressor.model.model) (class in neural_compressor.tensorflow.utils.model) model (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcherTF attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.JitBasicSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.RecipeSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.SelfMHASearcher attribute) (neural_compressor.compression.pruner.pruning.BasePruning attribute) (neural_compressor.compression.pruner.pruning.BasicPruning attribute) (neural_compressor.compression.pruner.pruning.RetrainFreePruning attribute) (neural_compressor.experimental.pytorch_pruner.pruning.Pruning attribute) model_slim() (in module neural_compressor.compression.pruner.model_slim.auto_slim) model_slim_ffn2() (in module neural_compressor.compression.pruner.model_slim.auto_slim) model_slim_mha() (in module neural_compressor.compression.pruner.model_slim.auto_slim) ModelConversion (class in neural_compressor.experimental.model_conversion) ModelSize (class in neural_compressor.objective) ModelWiseTuningSampler (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) module block_mask neural_compressor neural_compressor.adaptor neural_compressor.adaptor.adaptor neural_compressor.adaptor.keras neural_compressor.adaptor.keras_utils neural_compressor.adaptor.keras_utils.conv2d neural_compressor.adaptor.keras_utils.dense neural_compressor.adaptor.keras_utils.depthwise_conv2d neural_compressor.adaptor.keras_utils.pool2d neural_compressor.adaptor.keras_utils.quantizer neural_compressor.adaptor.keras_utils.separable_conv2d neural_compressor.adaptor.mxnet neural_compressor.adaptor.mxnet_utils neural_compressor.adaptor.mxnet_utils.util neural_compressor.adaptor.onnxrt neural_compressor.adaptor.ox_utils neural_compressor.adaptor.ox_utils.calibration neural_compressor.adaptor.ox_utils.calibrator neural_compressor.adaptor.ox_utils.operators neural_compressor.adaptor.ox_utils.operators.activation neural_compressor.adaptor.ox_utils.operators.argmax neural_compressor.adaptor.ox_utils.operators.attention neural_compressor.adaptor.ox_utils.operators.binary_op neural_compressor.adaptor.ox_utils.operators.concat neural_compressor.adaptor.ox_utils.operators.conv neural_compressor.adaptor.ox_utils.operators.direct_q8 neural_compressor.adaptor.ox_utils.operators.embed_layernorm neural_compressor.adaptor.ox_utils.operators.gather neural_compressor.adaptor.ox_utils.operators.gavgpool neural_compressor.adaptor.ox_utils.operators.gemm neural_compressor.adaptor.ox_utils.operators.lstm neural_compressor.adaptor.ox_utils.operators.matmul neural_compressor.adaptor.ox_utils.operators.maxpool neural_compressor.adaptor.ox_utils.operators.norm neural_compressor.adaptor.ox_utils.operators.ops neural_compressor.adaptor.ox_utils.operators.pad neural_compressor.adaptor.ox_utils.operators.pooling neural_compressor.adaptor.ox_utils.operators.reduce neural_compressor.adaptor.ox_utils.operators.resize neural_compressor.adaptor.ox_utils.operators.split neural_compressor.adaptor.ox_utils.operators.unary_op neural_compressor.adaptor.ox_utils.quantizer neural_compressor.adaptor.ox_utils.smooth_quant neural_compressor.adaptor.ox_utils.util neural_compressor.adaptor.ox_utils.weight_only neural_compressor.adaptor.pytorch neural_compressor.adaptor.query neural_compressor.adaptor.tensorflow neural_compressor.adaptor.tf_utils neural_compressor.adaptor.tf_utils.graph_converter neural_compressor.adaptor.tf_utils.graph_converter_without_calib neural_compressor.adaptor.tf_utils.graph_rewriter neural_compressor.adaptor.tf_utils.graph_rewriter.bf16 neural_compressor.adaptor.tf_utils.graph_rewriter.bf16.bf16_convert neural_compressor.adaptor.tf_utils.graph_rewriter.bf16.dequantize_cast_optimizer neural_compressor.adaptor.tf_utils.graph_rewriter.generic neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_add_to_biasadd neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_layout neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_leakyrelu neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_nan_to_random neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_placeholder_to_const neural_compressor.adaptor.tf_utils.graph_rewriter.generic.dilated_contraction neural_compressor.adaptor.tf_utils.graph_rewriter.generic.dummy_biasadd neural_compressor.adaptor.tf_utils.graph_rewriter.generic.expanddims_optimizer neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fetch_weight_from_reshape neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_batch_norm neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_constant neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_biasadd_add neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_column_wise_mul neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_conv_with_math neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_gelu neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_layer_norm neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_conv neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_fp32_conv neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_reshape_transpose neural_compressor.adaptor.tf_utils.graph_rewriter.generic.graph_cse_optimizer neural_compressor.adaptor.tf_utils.graph_rewriter.generic.grappler_pass neural_compressor.adaptor.tf_utils.graph_rewriter.generic.insert_print_node neural_compressor.adaptor.tf_utils.graph_rewriter.generic.move_squeeze_after_relu neural_compressor.adaptor.tf_utils.graph_rewriter.generic.pre_optimize neural_compressor.adaptor.tf_utils.graph_rewriter.generic.remove_training_nodes neural_compressor.adaptor.tf_utils.graph_rewriter.generic.rename_batch_norm neural_compressor.adaptor.tf_utils.graph_rewriter.generic.split_shared_input neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_equivalent_nodes neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_unused_nodes neural_compressor.adaptor.tf_utils.graph_rewriter.generic.switch_optimizer neural_compressor.adaptor.tf_utils.graph_rewriter.graph_base neural_compressor.adaptor.tf_utils.graph_rewriter.int8 neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_fake_quant neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_value neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_value_without_calib neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_conv_redundant_dequantize neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_conv_requantize neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_requantize neural_compressor.adaptor.tf_utils.graph_rewriter.int8.meta_op_optimizer neural_compressor.adaptor.tf_utils.graph_rewriter.int8.post_hostconst_converter neural_compressor.adaptor.tf_utils.graph_rewriter.int8.post_quantized_op_cse neural_compressor.adaptor.tf_utils.graph_rewriter.int8.rnn_convert neural_compressor.adaptor.tf_utils.graph_rewriter.int8.scale_propagation neural_compressor.adaptor.tf_utils.graph_rewriter.onnx neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_graph neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_node neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils neural_compressor.adaptor.tf_utils.graph_rewriter.qdq neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.insert_qdq_pattern neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.merge_duplicated_qdq neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.share_qdq_y_pattern neural_compressor.adaptor.tf_utils.graph_util neural_compressor.adaptor.tf_utils.quantize_graph neural_compressor.adaptor.tf_utils.quantize_graph.qat neural_compressor.adaptor.tf_utils.quantize_graph.qat.fake_quantize neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_config neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_helper neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.optimize_layer neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_add neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_base neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_bn neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_wrapper neural_compressor.adaptor.tf_utils.quantize_graph.qdq neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_bn neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_concatv2 neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_conv neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_deconv neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_in neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_matmul neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_pooling neural_compressor.adaptor.tf_utils.quantize_graph.qdq.optimize_qdq neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_base neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_bn neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_concatv2 neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_conv neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_for_intel_cpu neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_matmul neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_pooling neural_compressor.adaptor.tf_utils.quantize_graph_common neural_compressor.adaptor.tf_utils.smooth_quant_calibration neural_compressor.adaptor.tf_utils.smooth_quant_scaler neural_compressor.adaptor.tf_utils.tf2onnx_converter neural_compressor.adaptor.tf_utils.transform_graph neural_compressor.adaptor.tf_utils.transform_graph.bias_correction neural_compressor.adaptor.tf_utils.transform_graph.graph_transform_base neural_compressor.adaptor.tf_utils.transform_graph.insert_logging neural_compressor.adaptor.tf_utils.transform_graph.rerange_quantized_concat neural_compressor.adaptor.tf_utils.util neural_compressor.adaptor.torch_utils neural_compressor.adaptor.torch_utils.auto_round neural_compressor.adaptor.torch_utils.awq neural_compressor.adaptor.torch_utils.bf16_convert neural_compressor.adaptor.torch_utils.gptq neural_compressor.adaptor.torch_utils.hawq_metric neural_compressor.adaptor.torch_utils.layer_wise_quant neural_compressor.adaptor.torch_utils.layer_wise_quant.modified_pickle neural_compressor.adaptor.torch_utils.layer_wise_quant.quantize neural_compressor.adaptor.torch_utils.layer_wise_quant.torch_load neural_compressor.adaptor.torch_utils.layer_wise_quant.utils neural_compressor.adaptor.torch_utils.mixed_precision neural_compressor.adaptor.torch_utils.model_wrapper neural_compressor.adaptor.torch_utils.pattern_detector neural_compressor.adaptor.torch_utils.symbolic_trace neural_compressor.adaptor.torch_utils.teq neural_compressor.adaptor.torch_utils.util neural_compressor.adaptor.torch_utils.waq neural_compressor.adaptor.torch_utils.waq.auto_alpha neural_compressor.adaptor.torch_utils.waq.calibration neural_compressor.adaptor.torch_utils.waq.graph_trace neural_compressor.adaptor.torch_utils.waq.smooth_quant neural_compressor.adaptor.torch_utils.waq.utils neural_compressor.adaptor.torch_utils.weight_only 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.benchmark 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.compression neural_compressor.compression.callbacks neural_compressor.compression.distillation neural_compressor.compression.distillation.criterions neural_compressor.compression.distillation.optimizers neural_compressor.compression.distillation.utility neural_compressor.compression.hpo neural_compressor.compression.hpo.sa_optimizer neural_compressor.compression.hpo.search_algorithms neural_compressor.compression.hpo.search_space neural_compressor.compression.pruner neural_compressor.compression.pruner.criteria neural_compressor.compression.pruner.dsnot neural_compressor.compression.pruner.model_slim neural_compressor.compression.pruner.model_slim.auto_slim neural_compressor.compression.pruner.model_slim.pattern_analyzer neural_compressor.compression.pruner.model_slim.weight_slim neural_compressor.compression.pruner.patterns neural_compressor.compression.pruner.patterns.base neural_compressor.compression.pruner.patterns.mha neural_compressor.compression.pruner.patterns.ninm neural_compressor.compression.pruner.patterns.nxm neural_compressor.compression.pruner.pruners neural_compressor.compression.pruner.pruners.base neural_compressor.compression.pruner.pruners.basic neural_compressor.compression.pruner.pruners.block_mask neural_compressor.compression.pruner.pruners.mha neural_compressor.compression.pruner.pruners.pattern_lock neural_compressor.compression.pruner.pruners.progressive neural_compressor.compression.pruner.pruners.retrain_free neural_compressor.compression.pruner.pruners.sparse_gpt neural_compressor.compression.pruner.pruning neural_compressor.compression.pruner.regs neural_compressor.compression.pruner.schedulers neural_compressor.compression.pruner.tf_criteria neural_compressor.compression.pruner.utils neural_compressor.compression.pruner.wanda neural_compressor.compression.pruner.wanda.prune neural_compressor.compression.pruner.wanda.utils neural_compressor.compression.pruner.wanda.wrapper neural_compressor.conf neural_compressor.conf.config neural_compressor.conf.dotdict neural_compressor.conf.pythonic_config neural_compressor.config neural_compressor.contrib neural_compressor.contrib.strategy neural_compressor.contrib.strategy.sigopt neural_compressor.contrib.strategy.tpe neural_compressor.data neural_compressor.data.dataloaders neural_compressor.data.dataloaders.base_dataloader neural_compressor.data.dataloaders.dataloader neural_compressor.data.dataloaders.default_dataloader neural_compressor.data.dataloaders.fetcher neural_compressor.data.dataloaders.mxnet_dataloader neural_compressor.data.dataloaders.onnxrt_dataloader neural_compressor.data.dataloaders.pytorch_dataloader neural_compressor.data.dataloaders.sampler neural_compressor.data.dataloaders.tensorflow_dataloader 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.coco_transform neural_compressor.data.transforms.imagenet_transform neural_compressor.data.transforms.postprocess neural_compressor.data.transforms.tokenization neural_compressor.data.transforms.transform neural_compressor.experimental neural_compressor.experimental.benchmark neural_compressor.experimental.common neural_compressor.experimental.common.criterion neural_compressor.experimental.common.dataloader neural_compressor.experimental.common.metric neural_compressor.experimental.common.model neural_compressor.experimental.common.optimizer neural_compressor.experimental.common.postprocess neural_compressor.experimental.common.torch_utils neural_compressor.experimental.component neural_compressor.experimental.compression neural_compressor.experimental.contrib neural_compressor.experimental.contrib.strategy neural_compressor.experimental.contrib.strategy.sigopt neural_compressor.experimental.contrib.strategy.tpe neural_compressor.experimental.data neural_compressor.experimental.data.dataloaders neural_compressor.experimental.data.dataloaders.base_dataloader neural_compressor.experimental.data.dataloaders.dataloader neural_compressor.experimental.data.dataloaders.default_dataloader neural_compressor.experimental.data.dataloaders.fetcher neural_compressor.experimental.data.dataloaders.mxnet_dataloader neural_compressor.experimental.data.dataloaders.onnxrt_dataloader neural_compressor.experimental.data.dataloaders.pytorch_dataloader neural_compressor.experimental.data.dataloaders.sampler neural_compressor.experimental.data.dataloaders.tensorflow_dataloader neural_compressor.experimental.data.datasets neural_compressor.experimental.data.datasets.bert_dataset neural_compressor.experimental.data.datasets.coco_dataset neural_compressor.experimental.data.datasets.dataset neural_compressor.experimental.data.datasets.dummy_dataset neural_compressor.experimental.data.datasets.dummy_dataset_v2 neural_compressor.experimental.data.datasets.imagenet_dataset neural_compressor.experimental.data.datasets.style_transfer_dataset neural_compressor.experimental.data.filters neural_compressor.experimental.data.filters.coco_filter neural_compressor.experimental.data.filters.filter neural_compressor.experimental.data.transforms neural_compressor.experimental.data.transforms.imagenet_transform neural_compressor.experimental.data.transforms.tokenization neural_compressor.experimental.data.transforms.transform neural_compressor.experimental.distillation neural_compressor.experimental.export neural_compressor.experimental.export.qlinear2qdq neural_compressor.experimental.export.tf2onnx neural_compressor.experimental.export.torch2onnx neural_compressor.experimental.graph_optimization neural_compressor.experimental.metric neural_compressor.experimental.metric.bleu neural_compressor.experimental.metric.bleu_util neural_compressor.experimental.metric.coco_label_map neural_compressor.experimental.metric.coco_tools neural_compressor.experimental.metric.evaluate_squad neural_compressor.experimental.metric.f1 neural_compressor.experimental.metric.metric neural_compressor.experimental.mixed_precision neural_compressor.experimental.model_conversion neural_compressor.experimental.nas neural_compressor.experimental.nas.basic_nas neural_compressor.experimental.nas.dynas neural_compressor.experimental.nas.nas neural_compressor.experimental.nas.nas_utils neural_compressor.experimental.nas.search_algorithms neural_compressor.experimental.pruner_legacy neural_compressor.experimental.pruner_legacy.gradient_sensitivity neural_compressor.experimental.pruner_legacy.group_lasso neural_compressor.experimental.pruner_legacy.magnitude neural_compressor.experimental.pruner_legacy.pattern_lock neural_compressor.experimental.pruner_legacy.pruner neural_compressor.experimental.pruning neural_compressor.experimental.pruning_recipes neural_compressor.experimental.pruning_recipes.patterns neural_compressor.experimental.pruning_recipes.patterns.pattern neural_compressor.experimental.pruning_recipes.patterns.tile_pattern neural_compressor.experimental.pruning_v2 neural_compressor.experimental.pytorch_pruner neural_compressor.experimental.pytorch_pruner.logger neural_compressor.experimental.pytorch_pruner.patterns neural_compressor.experimental.pytorch_pruner.prune_utils neural_compressor.experimental.pytorch_pruner.pruner neural_compressor.experimental.pytorch_pruner.pruning neural_compressor.experimental.pytorch_pruner.scheduler neural_compressor.experimental.quantization neural_compressor.experimental.scheduler neural_compressor.experimental.strategy neural_compressor.experimental.strategy.auto_mixed_precision neural_compressor.experimental.strategy.basic neural_compressor.experimental.strategy.bayesian neural_compressor.experimental.strategy.exhaustive neural_compressor.experimental.strategy.mse neural_compressor.experimental.strategy.mse_v2 neural_compressor.experimental.strategy.random neural_compressor.experimental.strategy.strategy neural_compressor.experimental.strategy.utils neural_compressor.experimental.strategy.utils.constant neural_compressor.experimental.strategy.utils.tuning_sampler neural_compressor.experimental.strategy.utils.tuning_space neural_compressor.experimental.strategy.utils.tuning_structs neural_compressor.experimental.strategy.utils.utility 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.mxnet_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.onnxrt neural_compressor.onnxrt.algorithms neural_compressor.onnxrt.algorithms.layer_wise neural_compressor.onnxrt.algorithms.layer_wise.core neural_compressor.onnxrt.algorithms.smoother neural_compressor.onnxrt.algorithms.smoother.calibrator neural_compressor.onnxrt.algorithms.smoother.core neural_compressor.onnxrt.algorithms.weight_only neural_compressor.onnxrt.algorithms.weight_only.awq neural_compressor.onnxrt.algorithms.weight_only.gptq neural_compressor.onnxrt.algorithms.weight_only.rtn neural_compressor.onnxrt.algorithms.weight_only.utility neural_compressor.onnxrt.quantization neural_compressor.onnxrt.quantization.algorithm_entry neural_compressor.onnxrt.quantization.autotune neural_compressor.onnxrt.quantization.calibrate neural_compressor.onnxrt.quantization.config neural_compressor.onnxrt.quantization.quantize neural_compressor.onnxrt.utils neural_compressor.onnxrt.utils.onnx_model neural_compressor.onnxrt.utils.utility neural_compressor.profiling neural_compressor.profiling.parser neural_compressor.profiling.parser.factory neural_compressor.profiling.parser.onnx_parser neural_compressor.profiling.parser.onnx_parser.factory neural_compressor.profiling.parser.onnx_parser.parser neural_compressor.profiling.parser.parser neural_compressor.profiling.parser.result neural_compressor.profiling.parser.tensorflow_parser neural_compressor.profiling.parser.tensorflow_parser.factory neural_compressor.profiling.parser.tensorflow_parser.parser neural_compressor.profiling.profiler neural_compressor.profiling.profiler.factory neural_compressor.profiling.profiler.onnxrt_profiler neural_compressor.profiling.profiler.onnxrt_profiler.factory neural_compressor.profiling.profiler.onnxrt_profiler.profiler neural_compressor.profiling.profiler.onnxrt_profiler.utils neural_compressor.profiling.profiler.profiler neural_compressor.profiling.profiler.tensorflow_profiler neural_compressor.profiling.profiler.tensorflow_profiler.factory neural_compressor.profiling.profiler.tensorflow_profiler.profiler neural_compressor.profiling.profiler.tensorflow_profiler.utils 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.template neural_compressor.template.api_doc_example 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.quantizer 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_converter_without_calib 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.freeze_value_without_calib 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.rnn_convert 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.qat neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_config neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_helper neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.optimize_layer neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_add neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_base neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_bn neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper 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.nets_factory neural_compressor.tensorflow.utils.utility neural_compressor.torch neural_compressor.torch.algorithms neural_compressor.torch.algorithms.base_algorithm neural_compressor.torch.algorithms.habana_fp8 neural_compressor.torch.algorithms.habana_fp8.fp8_quant neural_compressor.torch.algorithms.habana_fp8.modules neural_compressor.torch.algorithms.habana_fp8.observer neural_compressor.torch.algorithms.habana_fp8.save_load neural_compressor.torch.algorithms.habana_fp8.scale neural_compressor.torch.algorithms.habana_fp8.tensor 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.pt2e_quant neural_compressor.torch.algorithms.pt2e_quant.core 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.autoround 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.quant_api neural_compressor.torch.algorithms.weight_only.hqq.quantizer neural_compressor.torch.algorithms.weight_only.hqq.utility neural_compressor.torch.algorithms.weight_only.modules neural_compressor.torch.algorithms.weight_only.rtn neural_compressor.torch.algorithms.weight_only.teq neural_compressor.torch.algorithms.weight_only.utility neural_compressor.torch.amp neural_compressor.torch.amp.autocast neural_compressor.torch.amp.fp8 neural_compressor.torch.amp.fp8.functions neural_compressor.torch.quantization neural_compressor.torch.quantization.algorithm_entry neural_compressor.torch.quantization.autotune neural_compressor.torch.quantization.config neural_compressor.torch.quantization.load_entry neural_compressor.torch.quantization.quantize neural_compressor.torch.utils neural_compressor.torch.utils.auto_accelerator 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.kl_divergence neural_compressor.utils.load_huggingface neural_compressor.utils.logger neural_compressor.utils.neural_insights_utils neural_compressor.utils.options neural_compressor.utils.pytorch neural_compressor.utils.utility neural_compressor.utils.weights_details neural_compressor.version

module_debug_level1 (in module neural_compressor.template.api_doc_example) modules (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) move_input_device() (in module neural_compressor.adaptor.torch_utils.util) MoveSqueezeAfterReluOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.move_squeeze_after_relu) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu) MSE (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) mse (neural_compressor.experimental.metric.metric.RMSE attribute) (neural_compressor.metric.metric.RMSE attribute) mse_metric_gap() (in module neural_compressor.utils.utility) MSE_V2TuneStrategy (class in neural_compressor.experimental.strategy.mse_v2) (class in neural_compressor.strategy.mse_v2) MSETuneStrategy (class in neural_compressor.experimental.strategy.mse) (class in neural_compressor.strategy.mse) MulLinear (class in neural_compressor.adaptor.torch_utils.model_wrapper) (class in neural_compressor.torch.algorithms.weight_only.modules) MultiObjective (class in neural_compressor.objective) MXNet (class in neural_compressor.config) MxNetAdaptor (class in neural_compressor.adaptor.mxnet) MXNetCIFAR10 (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) MXNetCIFAR100 (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) MXNetCropResizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) MXNetCropToBoundingBox (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) MXNetDataLoader (class in neural_compressor.data.dataloaders.mxnet_dataloader) (class in neural_compressor.experimental.data.dataloaders.mxnet_dataloader) MXNetDatasets (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) MXNetFashionMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) MXNetFilters (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) MXNetImageFolder (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) MXNetImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) (class in neural_compressor.experimental.data.datasets.imagenet_dataset) MXNetMetrics (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) MXNetMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) MXNetModel (class in neural_compressor.model.mxnet_model) MXNetNormalizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) MXNetTransforms (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) MXNetTranspose (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform)

N

N (neural_compressor.compression.pruner.patterns.mha.PatternMHA attribute) (neural_compressor.compression.pruner.patterns.ninm.PytorchPatternNInM attribute) (neural_compressor.experimental.pytorch_pruner.patterns.PatternNInM attribute) NameCollector (class in neural_compressor.adaptor.mxnet_utils.util) NAS (class in neural_compressor.experimental.nas.nas) nas_registry() (in module neural_compressor.experimental.nas.nas_utils) NASBase (class in neural_compressor.experimental.nas.nas) NASConfig (class in neural_compressor.conf.config) (class in neural_compressor.config) ndarray_to_device() (in module neural_compressor.adaptor.mxnet_utils.util) neural_compressor module neural_compressor.adaptor module neural_compressor.adaptor.adaptor module neural_compressor.adaptor.keras module neural_compressor.adaptor.keras_utils module neural_compressor.adaptor.keras_utils.conv2d module neural_compressor.adaptor.keras_utils.dense module neural_compressor.adaptor.keras_utils.depthwise_conv2d module neural_compressor.adaptor.keras_utils.pool2d module neural_compressor.adaptor.keras_utils.quantizer module neural_compressor.adaptor.keras_utils.separable_conv2d module neural_compressor.adaptor.mxnet module neural_compressor.adaptor.mxnet_utils module neural_compressor.adaptor.mxnet_utils.util module neural_compressor.adaptor.onnxrt module neural_compressor.adaptor.ox_utils module neural_compressor.adaptor.ox_utils.calibration module neural_compressor.adaptor.ox_utils.calibrator module neural_compressor.adaptor.ox_utils.operators module neural_compressor.adaptor.ox_utils.operators.activation module neural_compressor.adaptor.ox_utils.operators.argmax module neural_compressor.adaptor.ox_utils.operators.attention module neural_compressor.adaptor.ox_utils.operators.binary_op module neural_compressor.adaptor.ox_utils.operators.concat module neural_compressor.adaptor.ox_utils.operators.conv module neural_compressor.adaptor.ox_utils.operators.direct_q8 module neural_compressor.adaptor.ox_utils.operators.embed_layernorm module neural_compressor.adaptor.ox_utils.operators.gather module neural_compressor.adaptor.ox_utils.operators.gavgpool module neural_compressor.adaptor.ox_utils.operators.gemm module neural_compressor.adaptor.ox_utils.operators.lstm module neural_compressor.adaptor.ox_utils.operators.matmul module neural_compressor.adaptor.ox_utils.operators.maxpool module neural_compressor.adaptor.ox_utils.operators.norm module neural_compressor.adaptor.ox_utils.operators.ops module neural_compressor.adaptor.ox_utils.operators.pad module neural_compressor.adaptor.ox_utils.operators.pooling module neural_compressor.adaptor.ox_utils.operators.reduce module neural_compressor.adaptor.ox_utils.operators.resize module neural_compressor.adaptor.ox_utils.operators.split module neural_compressor.adaptor.ox_utils.operators.unary_op module neural_compressor.adaptor.ox_utils.quantizer module neural_compressor.adaptor.ox_utils.smooth_quant module neural_compressor.adaptor.ox_utils.util module neural_compressor.adaptor.ox_utils.weight_only module neural_compressor.adaptor.pytorch module neural_compressor.adaptor.query module neural_compressor.adaptor.tensorflow module neural_compressor.adaptor.tf_utils module neural_compressor.adaptor.tf_utils.graph_converter module neural_compressor.adaptor.tf_utils.graph_converter_without_calib module neural_compressor.adaptor.tf_utils.graph_rewriter module neural_compressor.adaptor.tf_utils.graph_rewriter.bf16 module neural_compressor.adaptor.tf_utils.graph_rewriter.bf16.bf16_convert module neural_compressor.adaptor.tf_utils.graph_rewriter.bf16.dequantize_cast_optimizer module neural_compressor.adaptor.tf_utils.graph_rewriter.generic module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_add_to_biasadd module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_layout module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_leakyrelu module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_nan_to_random module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.convert_placeholder_to_const module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.dilated_contraction module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.dummy_biasadd module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.expanddims_optimizer module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fetch_weight_from_reshape module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_batch_norm module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_constant module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_biasadd_add module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_column_wise_mul module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_conv_with_math module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_gelu module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_layer_norm module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_conv module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_fp32_conv module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_reshape_transpose module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.graph_cse_optimizer module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.grappler_pass module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.insert_print_node module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.move_squeeze_after_relu module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.pre_optimize module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.remove_training_nodes module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.rename_batch_norm module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.split_shared_input module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_equivalent_nodes module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_unused_nodes module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.switch_optimizer module neural_compressor.adaptor.tf_utils.graph_rewriter.graph_base module neural_compressor.adaptor.tf_utils.graph_rewriter.int8 module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_fake_quant module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_value module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.freeze_value_without_calib module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_conv_redundant_dequantize module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_conv_requantize module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.fuse_matmul_requantize module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.meta_op_optimizer module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.post_hostconst_converter module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.post_quantized_op_cse module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.rnn_convert module neural_compressor.adaptor.tf_utils.graph_rewriter.int8.scale_propagation module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_graph module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_node module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils module neural_compressor.adaptor.tf_utils.graph_rewriter.qdq module neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.insert_qdq_pattern module neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.merge_duplicated_qdq module neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.share_qdq_y_pattern module neural_compressor.adaptor.tf_utils.graph_util module neural_compressor.adaptor.tf_utils.quantize_graph module neural_compressor.adaptor.tf_utils.quantize_graph.qat module neural_compressor.adaptor.tf_utils.quantize_graph.qat.fake_quantize module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_config module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_helper module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.optimize_layer module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_add module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_base module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_bn module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_wrapper module neural_compressor.adaptor.tf_utils.quantize_graph.qdq module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_bn module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_concatv2 module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_conv module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_deconv module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_in module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_matmul module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.fuse_qdq_pooling module neural_compressor.adaptor.tf_utils.quantize_graph.qdq.optimize_qdq module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_base module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_bn module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_concatv2 module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_conv module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_for_intel_cpu module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_matmul module neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_pooling module neural_compressor.adaptor.tf_utils.quantize_graph_common module neural_compressor.adaptor.tf_utils.smooth_quant_calibration module neural_compressor.adaptor.tf_utils.smooth_quant_scaler module neural_compressor.adaptor.tf_utils.tf2onnx_converter module neural_compressor.adaptor.tf_utils.transform_graph module neural_compressor.adaptor.tf_utils.transform_graph.bias_correction module neural_compressor.adaptor.tf_utils.transform_graph.graph_transform_base module neural_compressor.adaptor.tf_utils.transform_graph.insert_logging module neural_compressor.adaptor.tf_utils.transform_graph.rerange_quantized_concat module neural_compressor.adaptor.tf_utils.util module neural_compressor.adaptor.torch_utils module neural_compressor.adaptor.torch_utils.auto_round module neural_compressor.adaptor.torch_utils.awq module neural_compressor.adaptor.torch_utils.bf16_convert module neural_compressor.adaptor.torch_utils.gptq module neural_compressor.adaptor.torch_utils.hawq_metric module neural_compressor.adaptor.torch_utils.layer_wise_quant module neural_compressor.adaptor.torch_utils.layer_wise_quant.modified_pickle module neural_compressor.adaptor.torch_utils.layer_wise_quant.quantize module neural_compressor.adaptor.torch_utils.layer_wise_quant.torch_load module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils module neural_compressor.adaptor.torch_utils.mixed_precision module neural_compressor.adaptor.torch_utils.model_wrapper module neural_compressor.adaptor.torch_utils.pattern_detector module neural_compressor.adaptor.torch_utils.symbolic_trace module neural_compressor.adaptor.torch_utils.teq module neural_compressor.adaptor.torch_utils.util module neural_compressor.adaptor.torch_utils.waq module neural_compressor.adaptor.torch_utils.waq.auto_alpha module neural_compressor.adaptor.torch_utils.waq.calibration module neural_compressor.adaptor.torch_utils.waq.graph_trace module neural_compressor.adaptor.torch_utils.waq.smooth_quant module neural_compressor.adaptor.torch_utils.waq.utils module neural_compressor.adaptor.torch_utils.weight_only 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.benchmark 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.compression module neural_compressor.compression.callbacks module neural_compressor.compression.distillation module neural_compressor.compression.distillation.criterions module neural_compressor.compression.distillation.optimizers module neural_compressor.compression.distillation.utility module neural_compressor.compression.hpo module neural_compressor.compression.hpo.sa_optimizer module neural_compressor.compression.hpo.search_algorithms module neural_compressor.compression.hpo.search_space module neural_compressor.compression.pruner module neural_compressor.compression.pruner.criteria module neural_compressor.compression.pruner.dsnot module neural_compressor.compression.pruner.model_slim module neural_compressor.compression.pruner.model_slim.auto_slim module neural_compressor.compression.pruner.model_slim.pattern_analyzer module neural_compressor.compression.pruner.model_slim.weight_slim module neural_compressor.compression.pruner.patterns module neural_compressor.compression.pruner.patterns.base module neural_compressor.compression.pruner.patterns.mha module neural_compressor.compression.pruner.patterns.ninm module neural_compressor.compression.pruner.patterns.nxm module neural_compressor.compression.pruner.pruners module neural_compressor.compression.pruner.pruners.base module neural_compressor.compression.pruner.pruners.basic module neural_compressor.compression.pruner.pruners.block_mask module neural_compressor.compression.pruner.pruners.mha module neural_compressor.compression.pruner.pruners.pattern_lock module neural_compressor.compression.pruner.pruners.progressive module neural_compressor.compression.pruner.pruners.retrain_free module neural_compressor.compression.pruner.pruners.sparse_gpt module neural_compressor.compression.pruner.pruning module neural_compressor.compression.pruner.regs module neural_compressor.compression.pruner.schedulers module neural_compressor.compression.pruner.tf_criteria module neural_compressor.compression.pruner.utils module neural_compressor.compression.pruner.wanda module neural_compressor.compression.pruner.wanda.prune module neural_compressor.compression.pruner.wanda.utils module neural_compressor.compression.pruner.wanda.wrapper module neural_compressor.conf module neural_compressor.conf.config module neural_compressor.conf.dotdict module neural_compressor.conf.pythonic_config module neural_compressor.config module neural_compressor.contrib module neural_compressor.contrib.strategy module neural_compressor.contrib.strategy.sigopt module neural_compressor.contrib.strategy.tpe module neural_compressor.data module neural_compressor.data.dataloaders module neural_compressor.data.dataloaders.base_dataloader module neural_compressor.data.dataloaders.dataloader module neural_compressor.data.dataloaders.default_dataloader module neural_compressor.data.dataloaders.fetcher module neural_compressor.data.dataloaders.mxnet_dataloader module neural_compressor.data.dataloaders.onnxrt_dataloader module neural_compressor.data.dataloaders.pytorch_dataloader module neural_compressor.data.dataloaders.sampler module neural_compressor.data.dataloaders.tensorflow_dataloader 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.coco_transform 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.experimental module neural_compressor.experimental.benchmark module neural_compressor.experimental.common module neural_compressor.experimental.common.criterion module neural_compressor.experimental.common.dataloader module neural_compressor.experimental.common.metric module neural_compressor.experimental.common.model module neural_compressor.experimental.common.optimizer module neural_compressor.experimental.common.postprocess module neural_compressor.experimental.common.torch_utils module neural_compressor.experimental.component module neural_compressor.experimental.compression module neural_compressor.experimental.contrib module neural_compressor.experimental.contrib.strategy module neural_compressor.experimental.contrib.strategy.sigopt module neural_compressor.experimental.contrib.strategy.tpe module neural_compressor.experimental.data module neural_compressor.experimental.data.dataloaders module neural_compressor.experimental.data.dataloaders.base_dataloader module neural_compressor.experimental.data.dataloaders.dataloader module neural_compressor.experimental.data.dataloaders.default_dataloader module neural_compressor.experimental.data.dataloaders.fetcher module neural_compressor.experimental.data.dataloaders.mxnet_dataloader module neural_compressor.experimental.data.dataloaders.onnxrt_dataloader module neural_compressor.experimental.data.dataloaders.pytorch_dataloader module neural_compressor.experimental.data.dataloaders.sampler module neural_compressor.experimental.data.dataloaders.tensorflow_dataloader module neural_compressor.experimental.data.datasets module neural_compressor.experimental.data.datasets.bert_dataset module neural_compressor.experimental.data.datasets.coco_dataset module neural_compressor.experimental.data.datasets.dataset module neural_compressor.experimental.data.datasets.dummy_dataset module neural_compressor.experimental.data.datasets.dummy_dataset_v2 module neural_compressor.experimental.data.datasets.imagenet_dataset module neural_compressor.experimental.data.datasets.style_transfer_dataset module neural_compressor.experimental.data.filters module neural_compressor.experimental.data.filters.coco_filter module neural_compressor.experimental.data.filters.filter module neural_compressor.experimental.data.transforms module neural_compressor.experimental.data.transforms.imagenet_transform module neural_compressor.experimental.data.transforms.tokenization module neural_compressor.experimental.data.transforms.transform module neural_compressor.experimental.distillation module neural_compressor.experimental.export module neural_compressor.experimental.export.qlinear2qdq module neural_compressor.experimental.export.tf2onnx module neural_compressor.experimental.export.torch2onnx module neural_compressor.experimental.graph_optimization module neural_compressor.experimental.metric module neural_compressor.experimental.metric.bleu module neural_compressor.experimental.metric.bleu_util module neural_compressor.experimental.metric.coco_label_map module neural_compressor.experimental.metric.coco_tools module neural_compressor.experimental.metric.evaluate_squad module neural_compressor.experimental.metric.f1 module neural_compressor.experimental.metric.metric module neural_compressor.experimental.mixed_precision module neural_compressor.experimental.model_conversion module neural_compressor.experimental.nas module neural_compressor.experimental.nas.basic_nas module neural_compressor.experimental.nas.dynas module neural_compressor.experimental.nas.nas module

neural_compressor.experimental.nas.nas_utils module neural_compressor.experimental.nas.search_algorithms module neural_compressor.experimental.pruner_legacy module neural_compressor.experimental.pruner_legacy.gradient_sensitivity module neural_compressor.experimental.pruner_legacy.group_lasso module neural_compressor.experimental.pruner_legacy.magnitude module neural_compressor.experimental.pruner_legacy.pattern_lock module neural_compressor.experimental.pruner_legacy.pruner module neural_compressor.experimental.pruning module neural_compressor.experimental.pruning_recipes module neural_compressor.experimental.pruning_recipes.patterns module neural_compressor.experimental.pruning_recipes.patterns.pattern module neural_compressor.experimental.pruning_recipes.patterns.tile_pattern module neural_compressor.experimental.pruning_v2 module neural_compressor.experimental.pytorch_pruner module neural_compressor.experimental.pytorch_pruner.logger module neural_compressor.experimental.pytorch_pruner.patterns module neural_compressor.experimental.pytorch_pruner.prune_utils module neural_compressor.experimental.pytorch_pruner.pruner module neural_compressor.experimental.pytorch_pruner.pruning module neural_compressor.experimental.pytorch_pruner.scheduler module neural_compressor.experimental.quantization module neural_compressor.experimental.scheduler module neural_compressor.experimental.strategy module neural_compressor.experimental.strategy.auto_mixed_precision module neural_compressor.experimental.strategy.basic module neural_compressor.experimental.strategy.bayesian module neural_compressor.experimental.strategy.exhaustive module neural_compressor.experimental.strategy.mse module neural_compressor.experimental.strategy.mse_v2 module neural_compressor.experimental.strategy.random module neural_compressor.experimental.strategy.strategy module neural_compressor.experimental.strategy.utils module neural_compressor.experimental.strategy.utils.constant module neural_compressor.experimental.strategy.utils.tuning_sampler module neural_compressor.experimental.strategy.utils.tuning_space module neural_compressor.experimental.strategy.utils.tuning_structs module neural_compressor.experimental.strategy.utils.utility 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.mxnet_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.onnxrt module neural_compressor.onnxrt.algorithms module neural_compressor.onnxrt.algorithms.layer_wise module neural_compressor.onnxrt.algorithms.layer_wise.core module neural_compressor.onnxrt.algorithms.smoother module neural_compressor.onnxrt.algorithms.smoother.calibrator module neural_compressor.onnxrt.algorithms.smoother.core module neural_compressor.onnxrt.algorithms.weight_only module neural_compressor.onnxrt.algorithms.weight_only.awq module neural_compressor.onnxrt.algorithms.weight_only.gptq module neural_compressor.onnxrt.algorithms.weight_only.rtn module neural_compressor.onnxrt.algorithms.weight_only.utility module neural_compressor.onnxrt.quantization module neural_compressor.onnxrt.quantization.algorithm_entry module neural_compressor.onnxrt.quantization.autotune module neural_compressor.onnxrt.quantization.calibrate module neural_compressor.onnxrt.quantization.config module neural_compressor.onnxrt.quantization.quantize module neural_compressor.onnxrt.utils module neural_compressor.onnxrt.utils.onnx_model module neural_compressor.onnxrt.utils.utility module neural_compressor.profiling module neural_compressor.profiling.parser module neural_compressor.profiling.parser.factory module neural_compressor.profiling.parser.onnx_parser module neural_compressor.profiling.parser.onnx_parser.factory module neural_compressor.profiling.parser.onnx_parser.parser module neural_compressor.profiling.parser.parser module neural_compressor.profiling.parser.result module neural_compressor.profiling.parser.tensorflow_parser module neural_compressor.profiling.parser.tensorflow_parser.factory module neural_compressor.profiling.parser.tensorflow_parser.parser module neural_compressor.profiling.profiler module neural_compressor.profiling.profiler.factory module neural_compressor.profiling.profiler.onnxrt_profiler module neural_compressor.profiling.profiler.onnxrt_profiler.factory module neural_compressor.profiling.profiler.onnxrt_profiler.profiler module neural_compressor.profiling.profiler.onnxrt_profiler.utils module neural_compressor.profiling.profiler.profiler module neural_compressor.profiling.profiler.tensorflow_profiler module neural_compressor.profiling.profiler.tensorflow_profiler.factory module neural_compressor.profiling.profiler.tensorflow_profiler.profiler module neural_compressor.profiling.profiler.tensorflow_profiler.utils 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.template module neural_compressor.template.api_doc_example 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.quantizer 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_converter_without_calib 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.freeze_value_without_calib 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.rnn_convert 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.qat module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_config module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_helper module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.optimize_layer module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_add module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_base module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_bn module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper 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.nets_factory module neural_compressor.tensorflow.utils.utility module neural_compressor.torch module neural_compressor.torch.algorithms module neural_compressor.torch.algorithms.base_algorithm module neural_compressor.torch.algorithms.habana_fp8 module neural_compressor.torch.algorithms.habana_fp8.fp8_quant module neural_compressor.torch.algorithms.habana_fp8.modules module neural_compressor.torch.algorithms.habana_fp8.observer module neural_compressor.torch.algorithms.habana_fp8.save_load module neural_compressor.torch.algorithms.habana_fp8.scale module neural_compressor.torch.algorithms.habana_fp8.tensor 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.pt2e_quant module neural_compressor.torch.algorithms.pt2e_quant.core 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.autoround 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.quant_api module neural_compressor.torch.algorithms.weight_only.hqq.quantizer module neural_compressor.torch.algorithms.weight_only.hqq.utility module neural_compressor.torch.algorithms.weight_only.modules module neural_compressor.torch.algorithms.weight_only.rtn module neural_compressor.torch.algorithms.weight_only.teq module neural_compressor.torch.algorithms.weight_only.utility module neural_compressor.torch.amp module neural_compressor.torch.amp.autocast module neural_compressor.torch.amp.fp8 module neural_compressor.torch.amp.fp8.functions 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.load_entry module neural_compressor.torch.quantization.quantize module neural_compressor.torch.utils module neural_compressor.torch.utils.auto_accelerator 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.kl_divergence module neural_compressor.utils.load_huggingface module neural_compressor.utils.logger module neural_compressor.utils.neural_insights_utils 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_collector (class in neural_compressor.adaptor.torch_utils.hawq_metric) node_from_map() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_layer_norm) (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.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_layer_norm) (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.experimental.metric.bleu.UnicodeRegex attribute) (neural_compressor.metric.bleu.UnicodeRegex attribute) NormalizationOperator (class in neural_compressor.adaptor.ox_utils.operators.norm) normalize_answer() (in module neural_compressor.experimental.metric.f1) (in module neural_compressor.metric.f1) NormalizeTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) NormalizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) num_correct (neural_compressor.experimental.metric.metric.GeneralTopK attribute) (neural_compressor.experimental.metric.metric.TensorflowTopK attribute) (neural_compressor.metric.metric.GeneralTopK attribute) (neural_compressor.metric.metric.TensorflowTopK attribute) num_sample (neural_compressor.experimental.metric.metric.GeneralTopK attribute) (neural_compressor.experimental.metric.metric.TensorflowTopK attribute) (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) OneshotScheduler (class in neural_compressor.compression.pruner.schedulers) (class in neural_compressor.experimental.pytorch_pruner.scheduler) ONNX (class in neural_compressor.config) onnx_qlinear_to_qdq() (in module neural_compressor.experimental.export.qlinear2qdq) OnnxBilinearImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) OnnxGraph (class in neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_graph) ONNXModel (class in neural_compressor.model.onnx_model) (class in neural_compressor.onnxrt.utils.onnx_model) OnnxNode (class in neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_node) OnnxOpSchema (class in neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema) OnnxProfilingParser (class in neural_compressor.profiling.parser.onnx_parser.parser) ONNXQlinear2QDQConfig (class in neural_compressor.config) ONNXResizeCropImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) onnxrt (class in neural_compressor.utils.options) ONNXRT_IntegerOpsAdaptor (class in neural_compressor.adaptor.onnxrt) ONNXRT_QDQAdaptor (class in neural_compressor.adaptor.onnxrt) ONNXRT_QLinearOpsAdaptor (class in neural_compressor.adaptor.onnxrt) ONNXRT_WeightOnlyAdaptor (class in neural_compressor.adaptor.onnxrt) ONNXRTAugment (class in neural_compressor.adaptor.ox_utils.calibration) ONNXRTBertDataLoader (class in neural_compressor.data.dataloaders.onnxrt_dataloader) (class in neural_compressor.experimental.data.dataloaders.onnxrt_dataloader) ONNXRTBertDataset (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.experimental.data.datasets.bert_dataset) ONNXRTCropToBoundingBox (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ONNXRTDataLoader (class in neural_compressor.data.dataloaders.onnxrt_dataloader) (class in neural_compressor.experimental.data.dataloaders.onnxrt_dataloader) ONNXRTGLUE (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) ONNXRTImagenetDataset (class in neural_compressor.data.datasets.imagenet_dataset) (class in neural_compressor.experimental.data.datasets.imagenet_dataset) ONNXRTITDatasets (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset)

ONNXRTITFilters (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) ONNXRTITMetrics (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) ONNXRTITTransforms (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) OnnxrtParserFactory (class in neural_compressor.profiling.parser.onnx_parser.factory) ONNXRTQLDatasets (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) ONNXRTQLFilters (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) ONNXRTQLMetrics (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) ONNXRTQLTransforms (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ONNXRUNTIMEAdaptor (class in neural_compressor.adaptor.onnxrt) op_registry() (in module neural_compressor.adaptor.ox_utils.operators.ops) OpEntry (class in neural_compressor.utils.utility) Operator (class in neural_compressor.adaptor.ox_utils.operators.ops) OptimizedModel (class in neural_compressor.utils.load_huggingface) OptimizeQDQGraph (class in neural_compressor.adaptor.tf_utils.quantize_graph.qdq.optimize_qdq) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq) optimizer_registry() (in module neural_compressor.compression.distillation.optimizers) (in module neural_compressor.experimental.common.optimizer) Optimizers (class in neural_compressor.compression.distillation.optimizers) (class in neural_compressor.experimental.common.optimizer) Options (class in neural_compressor.conf.pythonic_config) (class in neural_compressor.config) OpTuningConfig (class in neural_compressor.experimental.strategy.utils.tuning_structs) (class in neural_compressor.strategy.utils.tuning_structs) OpType (class in neural_compressor.adaptor.mxnet_utils.util) OpTypeWiseTuningSampler (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) OpWiseTuningSampler (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) OrderedDefaultDict (class in neural_compressor.experimental.strategy.utils.utility) (class in neural_compressor.strategy.utils.utility) ORTSmoothQuant (class in neural_compressor.adaptor.ox_utils.smooth_quant)

P

pad_tensor() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.utility) PaddedCenterCropTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PadOperator (class in neural_compressor.adaptor.ox_utils.operators.pad) ParamLevel (class in neural_compressor.common.tuning_param) parse_auto_slim_config() (in module neural_compressor.compression.pruner.model_slim.auto_slim) parse_last_linear() (in module neural_compressor.compression.pruner.utils) parse_last_linear_tf() (in module neural_compressor.compression.pruner.utils) parse_not_to_prune() (in module neural_compressor.experimental.pytorch_pruner.prune_utils) parse_saved_model() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) parse_to_prune() (in module neural_compressor.compression.pruner.utils) (in module neural_compressor.experimental.pytorch_pruner.prune_utils) parse_to_prune_tf() (in module neural_compressor.compression.pruner.utils) parse_tune_config() (in module neural_compressor.adaptor.mxnet_utils.util) parse_valid_pruner_types() (in module neural_compressor.compression.pruner.pruners) ParseDecodeBert (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.experimental.data.datasets.bert_dataset) ParseDecodeCoco (class in neural_compressor.data.datasets.coco_dataset) (class in neural_compressor.experimental.data.datasets.coco_dataset) ParseDecodeCocoTransform (class in neural_compressor.data.transforms.coco_transform) ParseDecodeImagenet (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) ParseDecodeImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) ParseDecodeVocTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ParserFactory (class in neural_compressor.profiling.parser.factory) paser_cfgs() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) Pattern (class in neural_compressor.experimental.pytorch_pruner.patterns) pattern (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.compression.pruner.pruners.basic.KerasBasicPruner attribute) (neural_compressor.compression.pruner.pruners.basic.PytorchBasicPruner attribute) (neural_compressor.compression.pruner.pruners.block_mask.PytorchBlockMaskPruner attribute) (neural_compressor.compression.pruner.pruners.retrain_free.PytorchRetrainFreePruner attribute) (neural_compressor.compression.pruner.pruners.sparse_gpt.SparseGPTPruner attribute) (neural_compressor.experimental.pytorch_pruner.patterns.Pattern attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) pattern_registry() (in module neural_compressor.experimental.pruning_recipes.patterns.pattern) pattern_to_internal() (in module neural_compressor.experimental.strategy.utils.tuning_space) (in module neural_compressor.strategy.utils.tuning_space) pattern_to_path() (in module neural_compressor.experimental.strategy.utils.tuning_space) (in module neural_compressor.strategy.utils.tuning_space) PatternBase (class in neural_compressor.experimental.pruning_recipes.patterns.pattern) PatternLockPruner (class in neural_compressor.experimental.pruner_legacy.pattern_lock) (class in neural_compressor.experimental.pytorch_pruner.pruner) PatternMHA (class in neural_compressor.compression.pruner.patterns.mha) PatternNInM (class in neural_compressor.experimental.pytorch_pruner.patterns) PatternNxM (class in neural_compressor.experimental.pytorch_pruner.patterns) PATTERNS (class in neural_compressor.experimental.pruning_recipes.patterns.pattern) patterns (neural_compressor.experimental.pruning_recipes.patterns.pattern.PATTERNS attribute) PercentileCalibrator (class in neural_compressor.adaptor.ox_utils.calibrator) Performance (class in neural_compressor.objective) PickleError, [1] PicklingError, [1] PoolOperator (class in neural_compressor.adaptor.ox_utils.operators.pooling) PostCompressionUtils (class in neural_compressor.compression.pruner.model_slim.weight_slim) PostCseOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.post_quantized_op_cse) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse) PostHostConstConverter (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.post_hostconst_converter) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter) Postprocess (class in neural_compressor.data.transforms.postprocess) (class in neural_compressor.experimental.common.postprocess) PostTrainingQuantConfig (class in neural_compressor.config) pred_list (neural_compressor.experimental.metric.metric.Accuracy attribute) (neural_compressor.experimental.metric.metric.MAE attribute) (neural_compressor.experimental.metric.metric.MSE attribute) (neural_compressor.metric.metric.Accuracy attribute) (neural_compressor.metric.metric.MAE attribute) (neural_compressor.metric.metric.MSE attribute) predictions (neural_compressor.experimental.metric.bleu.BLEU attribute) (neural_compressor.metric.bleu.BLEU attribute) PreOptimization (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.pre_optimize) (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_dataloader() (in module neural_compressor.adaptor.mxnet_utils.util) prepare_inputs() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.utility) prepare_model() (in module neural_compressor.adaptor.mxnet_utils.util) prepare_model_data() (in module neural_compressor.adaptor.mxnet_utils.util) prepare_pruning() (in module neural_compressor.compression.pruner) preprocess_user_cfg() (in module neural_compressor.strategy.utils.utility) print_iterables() (in module neural_compressor.compression.pruner.model_slim.pattern_analyzer) print_op_list() (in module neural_compressor.utils.utility) print_table() (in module neural_compressor.utils.utility) process_and_check_config() (in module neural_compressor.compression.pruner.utils) (in module neural_compressor.experimental.pytorch_pruner.prune_utils) process_config() (in module neural_compressor.compression.pruner.utils) (in module neural_compressor.experimental.pytorch_pruner.prune_utils) process_weight_config() (in module neural_compressor.compression.pruner.utils) process_yaml_config() (in module neural_compressor.compression.pruner.utils) profile() (in module neural_compressor.benchmark) Profiler (class in neural_compressor.profiling.profiler.onnxrt_profiler.profiler) (class in neural_compressor.profiling.profiler.profiler) (class in neural_compressor.profiling.profiler.tensorflow_profiler.profiler) ProfilerFactory (class in neural_compressor.profiling.profiler.factory) (class in neural_compressor.profiling.profiler.onnxrt_profiler.factory) (class in neural_compressor.profiling.profiler.tensorflow_profiler.factory)

ProfilingParser (class in neural_compressor.profiling.parser.parser) ProfilingResult (class in neural_compressor.profiling.parser.result) prune_wanda() (in module neural_compressor.compression.pruner.wanda.prune) Pruner (class in neural_compressor.experimental.pruner_legacy.pruner) (class in neural_compressor.experimental.pytorch_pruner.pruner) pruner_info (neural_compressor.compression.pruner.pruning.BasePruning attribute) (neural_compressor.compression.pruner.pruning.BasicPruning attribute) (neural_compressor.compression.pruner.pruning.RetrainFreePruning attribute) (neural_compressor.experimental.pytorch_pruner.pruning.Pruning attribute) pruner_registry() (in module neural_compressor.experimental.pruner_legacy.pruner) pruners (neural_compressor.compression.pruner.pruning.BasePruning attribute) (neural_compressor.compression.pruner.pruning.BasicPruning attribute) (neural_compressor.compression.pruner.pruning.RetrainFreePruning attribute) (neural_compressor.experimental.pruning.Pruning attribute) (neural_compressor.experimental.pruning_v2.Pruning attribute) (neural_compressor.experimental.pytorch_pruner.pruning.Pruning attribute) PrunerV2 (class in neural_compressor.conf.config) Pruning (class in neural_compressor.experimental.pruning) (class in neural_compressor.experimental.pruning_v2) (class in neural_compressor.experimental.pytorch_pruner.pruning) Pruning_Conf (class in neural_compressor.conf.config) pruning_frequency (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) PruningCallbacks (class in neural_compressor.compression.callbacks) PruningCriterion (class in neural_compressor.compression.pruner.criteria) (class in neural_compressor.compression.pruner.tf_criteria) PruningScheduler (class in neural_compressor.compression.pruner.schedulers) punct_nondigit_re (neural_compressor.experimental.metric.bleu.UnicodeRegex attribute) (neural_compressor.metric.bleu.UnicodeRegex attribute) PythonMultiheadAttentionPruner (class in neural_compressor.compression.pruner.pruners.mha) PyTorch (class in neural_compressor.config) PyTorch_FXAdaptor (class in neural_compressor.adaptor.pytorch) PyTorch_IPEXAdaptor (class in neural_compressor.adaptor.pytorch) PyTorchAdaptor (class in neural_compressor.adaptor.pytorch) PyTorchAlignImageChannel (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PyTorchBaseModel (class in neural_compressor.model.torch_model) PytorchBasePattern (class in neural_compressor.compression.pruner.patterns.base) PytorchBasePruner (class in neural_compressor.compression.pruner.pruners.base) PytorchBasicPruner (class in neural_compressor.compression.pruner.pruners.basic) PytorchBertDataset (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.experimental.data.datasets.bert_dataset) PytorchBlockMaskPruner (class in neural_compressor.compression.pruner.pruners.block_mask) PytorchCIFAR10 (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) PytorchCIFAR100 (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) PyTorchCriterions (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) PyTorchCropResizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PyTorchCrossEntropyLoss (class in neural_compressor.compression.distillation.criterions) PyTorchDataLoader (class in neural_compressor.data.dataloaders.pytorch_dataloader) (class in neural_compressor.experimental.data.dataloaders.pytorch_dataloader) PyTorchDatasets (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) PytorchFashionMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) PyTorchFilters (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) PyTorchFXModel (class in neural_compressor.model.torch_model) PytorchImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) (class in neural_compressor.experimental.data.datasets.imagenet_dataset) PyTorchIntermediateLayersKnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) PyTorchIntermediateLayersKnowledgeDistillationLossWrapper (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) PyTorchKnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) PyTorchKnowledgeDistillationLossWrapper (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) PyTorchLoss (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) PyTorchMetrics (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) PytorchMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) PyTorchModel (class in neural_compressor.model.torch_model) PytorchMxnetTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PytorchMxnetWrapDataset (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) PytorchMxnetWrapFunction (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.datasets.dataset) (class in neural_compressor.experimental.data.transforms.transform) PyTorchNormalizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PyTorchOptimizers (class in neural_compressor.compression.distillation.optimizers) (class in neural_compressor.experimental.common.optimizer) PytorchPatternLockPruner (class in neural_compressor.compression.pruner.pruners.pattern_lock) PytorchPatternNInM (class in neural_compressor.compression.pruner.patterns.ninm) PytorchPatternNxM (class in neural_compressor.compression.pruner.patterns.nxm) PytorchProgressivePruner (class in neural_compressor.compression.pruner.pruners.progressive) PytorchRetrainFreePruner (class in neural_compressor.compression.pruner.pruners.retrain_free) PyTorchSelfKnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) PyTorchSelfKnowledgeDistillationLossWrapper (class in neural_compressor.compression.distillation.criterions) PyTorchSGD (class in neural_compressor.compression.distillation.optimizers) (class in neural_compressor.experimental.common.optimizer) PyTorchTransforms (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PyTorchTranspose (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) PyTorchWeightOnlyAdaptor (class in neural_compressor.adaptor.pytorch)

Q

QActivationOperator (class in neural_compressor.adaptor.ox_utils.operators.activation) QArgMaxOperator (class in neural_compressor.adaptor.ox_utils.operators.argmax) qat_clone_function() (in module neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_helper) (in module neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_helper) QAttentionOperator (class in neural_compressor.adaptor.ox_utils.operators.attention) QBinaryOperator (class in neural_compressor.adaptor.ox_utils.operators.binary_op) QConcatOperator (class in neural_compressor.adaptor.ox_utils.operators.concat) QConvOperator (class in neural_compressor.adaptor.ox_utils.operators.conv) QDirectOperator (class in neural_compressor.adaptor.ox_utils.operators.direct_q8) qdq_tensor() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.utility) qdq_weight_actor() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.utility) qdq_weight_asym() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.utility) qdq_weight_sym() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.utility) QEmbedLayerNormalizationOperator (class in neural_compressor.adaptor.ox_utils.operators.embed_layernorm) QGatherOperator (class in neural_compressor.adaptor.ox_utils.operators.gather) QGemmOperator (class in neural_compressor.adaptor.ox_utils.operators.gemm) QGlobalAveragePoolOperator (class in neural_compressor.adaptor.ox_utils.operators.gavgpool) QMatMulOperator (class in neural_compressor.adaptor.ox_utils.operators.matmul) QMaxPoolOperator (class in neural_compressor.adaptor.ox_utils.operators.maxpool) qop_registry() (in module neural_compressor.adaptor.ox_utils.operators.ops) QOperator (class in neural_compressor.adaptor.ox_utils.operators.ops) QPadOperator (class in neural_compressor.adaptor.ox_utils.operators.pad) QPoolOperator (class in neural_compressor.adaptor.ox_utils.operators.pooling) QResizeOperator (class in neural_compressor.adaptor.ox_utils.operators.resize) QSplitOperator (class in neural_compressor.adaptor.ox_utils.operators.split) quant_dequant_data() (in module neural_compressor.adaptor.ox_utils.smooth_quant) quant_mode_from_pattern() (in module neural_compressor.experimental.strategy.utils.tuning_space) (in module neural_compressor.strategy.utils.tuning_space) quant_tensor() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.utility) (in module neural_compressor.torch.algorithms.weight_only.utility) quant_weight() (in module neural_compressor.adaptor.torch_utils.weight_only) quant_weight_w_scale() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.utility) QuantFormat (class in neural_compressor.adaptor.ox_utils.util) Quantization (class in neural_compressor.experimental.quantization) Quantization_Conf (class in neural_compressor.conf.config) QuantizationAwareTrainingCallbacks (class in neural_compressor.compression.callbacks) QuantizationAwareTrainingConfig (class in neural_compressor.config) QuantizationConfig (class in neural_compressor.conf.pythonic_config) QuantizationMode (class in neural_compressor.adaptor.ox_utils.util) quantize() (in module neural_compressor.adaptor.torch_utils.gptq) (in module neural_compressor.torch.algorithms.weight_only.gptq) (in module neural_compressor.torch.quantization.quantize)

quantize_4bit() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.utility) quantize_data() (in module neural_compressor.adaptor.ox_utils.util) (in module neural_compressor.onnxrt.utils.utility) quantize_data_per_channel() (in module neural_compressor.adaptor.ox_utils.util) quantize_data_with_scale_zero() (in module neural_compressor.adaptor.ox_utils.util) quantize_model() (in module neural_compressor.tensorflow.quantization.quantize) quantize_model_with_single_config() (in module neural_compressor.tensorflow.quantization.quantize) quantize_nparray() (in module neural_compressor.adaptor.ox_utils.util) quantize_sym_model() (in module neural_compressor.adaptor.mxnet_utils.util) QuantizeConfig (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_config) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_config) QuantizedInitializer (class in neural_compressor.adaptor.ox_utils.util) QuantizedInput (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) QuantizedRNNConverter (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.rnn_convert) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.rnn_convert) QuantizedValue (class in neural_compressor.adaptor.ox_utils.util) QuantizedValueType (class in neural_compressor.adaptor.ox_utils.util) QuantizeGraphBase (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_base) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base) QuantizeGraphForIntel (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_for_intel_cpu) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu) QuantizeGraphHelper (class in neural_compressor.adaptor.tf_utils.quantize_graph_common) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph_common) QuantizeLayerAdd (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_add) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_add) QuantizeLayerBase (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_base) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_base) QuantizeLayerBatchNormalization (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_layers.quantize_layer_bn) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_layers.quantize_layer_bn) QuantizeNodeBase (class in neural_compressor.adaptor.tf_utils.quantize_graph.quantize_graph_base) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base) Quantizer (class in neural_compressor.adaptor.ox_utils.quantizer) (class in neural_compressor.torch.algorithms.base_algorithm) QuantizeWrapper (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_wrapper) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper) QuantizeWrapperBase (class in neural_compressor.adaptor.tf_utils.quantize_graph.qat.quantize_wrapper) (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper) QuantOptions (class in neural_compressor.strategy.utils.utility) QuantType (class in neural_compressor.adaptor.ox_utils.util) (class in neural_compressor.strategy.utils.utility) query_quantizable_nodes() (in module neural_compressor.adaptor.mxnet_utils.util) QueryBackendCapability (class in neural_compressor.adaptor.query)

R

RandomCropTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomCropTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomHorizontalFlip (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomResizedCropMXNetTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomResizedCropPytorchTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomResizedCropTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomResizedCropTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RandomSearcher (class in neural_compressor.compression.hpo.search_algorithms) (class in neural_compressor.experimental.nas.search_algorithms) RandomTuneStrategy (class in neural_compressor.experimental.strategy.random) (class in neural_compressor.strategy.random) RandomVerticalFlip (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) read_graph() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) read_squad_examples() (in module neural_compressor.data.transforms.transform) (in module neural_compressor.experimental.data.transforms.transform) read_tensorflow_node_attrs() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) recipe (neural_compressor.compression.pruner.model_slim.pattern_analyzer.RecipeSearcher attribute) RecipeSearcher (class in neural_compressor.compression.pruner.model_slim.pattern_analyzer) reconstruct_saved_model() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) record_output() (in module neural_compressor.compression.distillation.utility) (in module neural_compressor.experimental.common.torch_utils) recover() (in module neural_compressor.utils.utility) recover_model_from_json() (in module neural_compressor.torch.algorithms.smooth_quant.save_load) (in module neural_compressor.utils.pytorch) ReduceMinMaxOperator (class in neural_compressor.adaptor.ox_utils.operators.reduce) ReduceOperator (class in neural_compressor.adaptor.ox_utils.operators.reduce) reg (neural_compressor.compression.pruner.pruners.basic.KerasBasicPruner attribute) (neural_compressor.compression.pruner.pruners.basic.PytorchBasicPruner attribute) (neural_compressor.compression.pruner.pruners.block_mask.PytorchBlockMaskPruner attribute) (neural_compressor.compression.pruner.pruners.retrain_free.PytorchRetrainFreePruner attribute) (neural_compressor.compression.pruner.pruners.sparse_gpt.SparseGPTPruner attribute) reg_terms (neural_compressor.compression.pruner.regs.GroupLasso attribute) register_accelerator() (in module neural_compressor.torch.utils.auto_accelerator) register_algo() (in module neural_compressor.onnxrt.utils.utility) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.torch.utils.utility) register_autotune() (in module neural_compressor.adaptor.torch_utils.waq.utils) (in module neural_compressor.torch.algorithms.smooth_quant.utility) register_config() (in module neural_compressor.common.base_config) register_criterion() (in module neural_compressor.compression.pruner.criteria) (in module neural_compressor.compression.pruner.tf_criteria) register_customer_metric() (in module neural_compressor.metric.metric) register_neural_insights_workload() (in module neural_compressor.utils.neural_insights_utils) register_pattern() (in module neural_compressor.compression.pruner.patterns.base) (in module neural_compressor.experimental.pytorch_pruner.patterns) register_pruner() (in module neural_compressor.compression.pruner.pruners.base) register_pruners() (in module neural_compressor.experimental.pytorch_pruner.pruner) register_pruning() (in module neural_compressor.compression.pruner.pruning) register_reg() (in module neural_compressor.compression.pruner.regs)

register_scheduler() (in module neural_compressor.compression.pruner.schedulers) (in module neural_compressor.experimental.pytorch_pruner.scheduler) register_searcher() (in module neural_compressor.compression.hpo.search_algorithms) register_searchspace() (in module neural_compressor.compression.hpo.search_space) register_supported_configs_for_fwk() (in module neural_compressor.common.base_config) RemovableActivationOperator (class in neural_compressor.adaptor.ox_utils.operators.activation) remove_init_from_model_input() (in module neural_compressor.adaptor.ox_utils.util) RemoveTrainingNodesOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.remove_training_nodes) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes) RenameBatchNormOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.rename_batch_norm) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm) RerangeQuantizedConcat (class in neural_compressor.adaptor.tf_utils.transform_graph.rerange_quantized_concat) (class in neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat) RescaleKerasPretrainTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RescaleTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RescaleTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) reset_non_value_to_default() (in module neural_compressor.experimental.pytorch_pruner.prune_utils) reset_none_to_default() (in module neural_compressor.compression.pruner.utils) reshape_in_channel_to_last() (in module neural_compressor.adaptor.torch_utils.waq.utils) (in module neural_compressor.torch.algorithms.smooth_quant.utility) reshape_scale_as_input() (in module neural_compressor.adaptor.torch_utils.waq.utils) (in module neural_compressor.torch.algorithms.smooth_quant.utility) reshape_scale_as_weight() (in module neural_compressor.adaptor.torch_utils.waq.utils) (in module neural_compressor.torch.algorithms.smooth_quant.utility) ResizeMXNetTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ResizeOperator (class in neural_compressor.adaptor.ox_utils.operators.resize) ResizePytorchTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ResizeTFTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ResizeTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ResizeWithAspectRatio (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) ResizeWithRatio (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) RetrainFreeCriterion (class in neural_compressor.compression.pruner.criteria) RetrainFreePruning (class in neural_compressor.compression.pruner.pruning) return_reorder_indice() (in module neural_compressor.compression.pruner.dsnot) reverted_data_type() (in module neural_compressor.experimental.strategy.utils.utility) (in module neural_compressor.strategy.utils.utility) RMSE (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) ROC (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) rtn_entry() (in module neural_compressor.torch.quantization.algorithm_entry) rtn_quantize() (in module neural_compressor.adaptor.ox_utils.weight_only) (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.onnxrt.algorithms.weight_only.rtn) (in module neural_compressor.torch.algorithms.weight_only.rtn) rtn_quantize_entry() (in module neural_compressor.onnxrt.quantization.algorithm_entry) RTNConfig (class in neural_compressor.onnxrt.quantization.config) (class in neural_compressor.torch.quantization.config) run_forward() (in module neural_compressor.adaptor.mxnet_utils.util) run_instance() (in module neural_compressor.benchmark)

S

sample (neural_compressor.experimental.metric.metric.Accuracy attribute) (neural_compressor.experimental.metric.metric.Loss attribute) (neural_compressor.metric.metric.Accuracy attribute) (neural_compressor.metric.metric.Loss attribute) Sampler (class in neural_compressor.data.dataloaders.sampler) (class in neural_compressor.experimental.data.dataloaders.sampler) save() (in module neural_compressor.compression.pruner) save_config_mapping() (in module neural_compressor.common.utils.save_load) save_for_huggingface_upstream() (in module neural_compressor.utils.load_huggingface) save_protobuf() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) saved_model_session() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) ScaleProPagationTransformer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.int8.scale_propagation) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation) Scheduler (class in neural_compressor.experimental.pytorch_pruner.scheduler) (class in neural_compressor.experimental.scheduler) scheduler (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.compression.pruner.pruners.basic.KerasBasicPruner attribute) (neural_compressor.compression.pruner.pruners.basic.PytorchBasicPruner attribute) (neural_compressor.compression.pruner.pruners.block_mask.PytorchBlockMaskPruner attribute) (neural_compressor.compression.pruner.pruners.retrain_free.PytorchRetrainFreePruner attribute) (neural_compressor.compression.pruner.pruners.sparse_gpt.SparseGPTPruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) scores (neural_compressor.compression.pruner.criteria.BlockMaskCriterion attribute) (neural_compressor.compression.pruner.criteria.GradientCriterion attribute) (neural_compressor.compression.pruner.criteria.MagnitudeCriterion attribute) (neural_compressor.compression.pruner.criteria.PruningCriterion attribute) (neural_compressor.compression.pruner.criteria.RetrainFreeCriterion attribute) (neural_compressor.compression.pruner.criteria.SnipCriterion attribute) (neural_compressor.compression.pruner.criteria.SnipMomentumCriterion attribute) (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.compression.pruner.tf_criteria.MagnitudeCriterion attribute) (neural_compressor.compression.pruner.tf_criteria.PruningCriterion attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) search_clip() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.utility) Searcher (class in neural_compressor.compression.hpo.search_algorithms) (class in neural_compressor.experimental.nas.search_algorithms) searching_results (neural_compressor.compression.pruner.model_slim.pattern_analyzer.JitBasicSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.RecipeSearcher attribute) SearchSpace (class in neural_compressor.compression.hpo.search_space) SelfKnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) SelfKnowledgeDistillationLossConfig (class in neural_compressor.config) SelfMHASearcher (class in neural_compressor.compression.pruner.model_slim.pattern_analyzer) SeqType (class in neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) SequentialSampler (class in neural_compressor.common.base_tuning) (class in neural_compressor.data.dataloaders.sampler) (class in neural_compressor.experimental.data.dataloaders.sampler) (class in neural_compressor.tensorflow.utils.data) set_all_env_var() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) set_backend() (in module neural_compressor.experimental.common.model) set_eager_execution() (in module neural_compressor.profiling.profiler.tensorflow_profiler.utils) set_env_var() (in module neural_compressor.benchmark) (in module neural_compressor.experimental.benchmark) set_module() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.adaptor.torch_utils.waq.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) set_name() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.tf2onnx_utils) 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) ShareQDQForItexYPatternOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.qdq.share_qdq_y_pattern) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern) show_memory_info() (in module neural_compressor.utils.utility) SigOptTuneStrategy (class in neural_compressor.contrib.strategy.sigopt) (class in neural_compressor.experimental.contrib.strategy.sigopt)

simple_inference() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.static_quant.utility) simple_progress_bar() (in module neural_compressor.adaptor.ox_utils.util) (in module neural_compressor.onnxrt.utils.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) SmoohQuantConfig (class in neural_compressor.onnxrt.quantization.config) smooth_distribution() (in module neural_compressor.adaptor.ox_utils.calibrator) smooth_quant_entry() (in module neural_compressor.onnxrt.quantization.algorithm_entry) smooth_quantize() (in module neural_compressor.torch.algorithms.smooth_quant.smooth_quant) Smoother (class in neural_compressor.onnxrt.algorithms.smoother.core) SmoothQuant (class in neural_compressor.algorithm.smooth_quant) (class in neural_compressor.tensorflow.algorithms.smoother.core) SmoothQuantCalibration (class in neural_compressor.adaptor.tf_utils.smooth_quant_calibration) (class in neural_compressor.tensorflow.algorithms.smoother.calibration) SmoothQuantCalibrationLLM (class in neural_compressor.adaptor.tf_utils.smooth_quant_calibration) (class in neural_compressor.tensorflow.algorithms.smoother.calibration) SmoothQuantConfig (class in neural_compressor.tensorflow.quantization.config) SmoothQuantSampler (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) SmoothQuantScaler (class in neural_compressor.adaptor.tf_utils.smooth_quant_scaler) (class in neural_compressor.tensorflow.algorithms.smoother.scaler) SmoothQuantScalerLLM (class in neural_compressor.adaptor.tf_utils.smooth_quant_scaler) (class in neural_compressor.tensorflow.algorithms.smoother.scaler) SnipCriterion (class in neural_compressor.compression.pruner.criteria) SnipMomentumCriterion (class in neural_compressor.compression.pruner.criteria) SnipMomentumPruner (class in neural_compressor.experimental.pytorch_pruner.pruner) SnipPruner (class in neural_compressor.experimental.pytorch_pruner.pruner) SparseDummyDataset (class in neural_compressor.data.datasets.dummy_dataset_v2) (class in neural_compressor.experimental.data.datasets.dummy_dataset_v2) SparseGPTPruner (class in neural_compressor.compression.pruner.pruners.sparse_gpt) SparseGPTPruning (class in neural_compressor.compression.pruner.pruning) split_shared_bias() (in module neural_compressor.adaptor.ox_utils.util) SplitOperator (class in neural_compressor.adaptor.ox_utils.operators.split) SplitSharedInputOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.split_shared_input) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input) SquadExample (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) SquadF1 (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) start_step (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) static_graph (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.ClassifierHeadSearcherTF attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.JitBasicSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.SelfMHASearcher attribute) static_quant_entry() (in module neural_compressor.tensorflow.quantization.algorithm_entry) static_quant_export() (in module neural_compressor.experimental.export.torch2onnx) StaticQuantConfig (class in neural_compressor.tensorflow.keras.quantization.config) (class in neural_compressor.tensorflow.quantization.config) StaticQuantQuantizer (class in neural_compressor.torch.algorithms.static_quant.static_quant) Statistics (class in neural_compressor.tensorflow.utils.utility) (class in neural_compressor.torch.algorithms.static_quant.utility) (class in neural_compressor.utils.utility) str2array() (in module neural_compressor.utils.utility) strategy_registry() (in module neural_compressor.experimental.strategy.strategy) (in module neural_compressor.strategy.strategy) strip_equivalent_nodes() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) strip_unused_nodes() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility) StripEquivalentNodesOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_equivalent_nodes) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes) StripUnusedNodesOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_unused_nodes) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes) StyleTransferDataset (class in neural_compressor.data.datasets.style_transfer_dataset) (class in neural_compressor.experimental.data.datasets.style_transfer_dataset) sum (neural_compressor.experimental.metric.metric.Loss attribute) (neural_compressor.metric.metric.Loss attribute) summary_benchmark() (in module neural_compressor.benchmark) SwitchOptimizer (class in neural_compressor.adaptor.tf_utils.graph_rewriter.generic.switch_optimizer) (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer) symbol_re (neural_compressor.experimental.metric.bleu.UnicodeRegex attribute) (neural_compressor.metric.bleu.UnicodeRegex attribute) symbolic_trace() (in module neural_compressor.adaptor.torch_utils.symbolic_trace)

T

target_layers (neural_compressor.compression.pruner.model_slim.pattern_analyzer.JitBasicSearcher attribute) (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) target_op_lut (neural_compressor.compression.pruner.model_slim.pattern_analyzer.Linear2LinearSearcher attribute) target_sparsity (neural_compressor.compression.pruner.patterns.base.BasePattern attribute) (neural_compressor.compression.pruner.patterns.base.KerasBasePattern attribute) (neural_compressor.compression.pruner.patterns.base.PytorchBasePattern attribute) target_sparsity_ratio (neural_compressor.compression.pruner.pruners.base.BasePruner attribute) (neural_compressor.compression.pruner.pruners.base.KerasBasePruner attribute) (neural_compressor.compression.pruner.pruners.base.PytorchBasePruner attribute) (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute) targets (neural_compressor.compression.pruner.model_slim.pattern_analyzer.RecipeSearcher attribute) TargetSpace (class in neural_compressor.experimental.strategy.bayesian) (class in neural_compressor.strategy.bayesian) TemplateAdaptor (class in neural_compressor.adaptor.pytorch) TensorCollector (class in neural_compressor.adaptor.mxnet_utils.util) TensorFlow (class in neural_compressor.config) Tensorflow (class in neural_compressor.data.datasets.dataset) Tensorflow_ITEXAdaptor (class in neural_compressor.adaptor.tensorflow) (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorFlowAdam (class in neural_compressor.compression.distillation.optimizers) TensorFlowAdamW (class in neural_compressor.compression.distillation.optimizers) (class in neural_compressor.experimental.common.optimizer) TensorFlowAdaptor (class in neural_compressor.adaptor.tensorflow) (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) TensorflowBertDataLoader (class in neural_compressor.data.dataloaders.tensorflow_dataloader) (class in neural_compressor.experimental.data.dataloaders.tensorflow_dataloader) TensorflowBertDataset (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.experimental.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) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowCIFAR100 (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowCOCOMAP (class in neural_compressor.experimental.metric.metric) (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) TensorflowCriterions (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) TensorflowCropToBoundingBox (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TensorFlowCrossEntropyLoss (class in neural_compressor.compression.distillation.criterions) TensorflowDataLoader (class in neural_compressor.data.dataloaders.tensorflow_dataloader) (class in neural_compressor.experimental.data.dataloaders.tensorflow_dataloader) TensorflowDatasets (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowFashionMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowFilters (class in neural_compressor.data.filters.filter) (class in neural_compressor.experimental.data.filters.filter) TensorflowImageFolder (class in neural_compressor.experimental.data.datasets.dataset) TensorflowImagenetDataset (class in neural_compressor.data.datasets.imagenet_dataset) (class in neural_compressor.experimental.data.datasets.imagenet_dataset) TensorflowImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset) (class in neural_compressor.experimental.data.datasets.imagenet_dataset) TensorflowImageRecord (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowKnowledgeDistillationLoss (class in neural_compressor.compression.distillation.criterions) TensorflowKnowledgeDistillationLossExternal (class in neural_compressor.compression.distillation.criterions) (class in neural_compressor.experimental.common.criterion) TensorflowKnowledgeDistillationLossWrapper (class in neural_compressor.compression.distillation.criterions) TensorflowLLMModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowMAP (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) TensorflowMetrics (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) TensorflowMNIST (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowModelZooBertDataLoader (class in neural_compressor.data.dataloaders.tensorflow_dataloader) (class in neural_compressor.experimental.data.dataloaders.tensorflow_dataloader) TensorflowModelZooBertDataset (class in neural_compressor.data.datasets.bert_dataset) (class in neural_compressor.experimental.data.datasets.bert_dataset) TensorflowOptimizers (class in neural_compressor.compression.distillation.optimizers) (class in neural_compressor.experimental.common.optimizer) TensorFlowParserFactory (class in neural_compressor.profiling.parser.tensorflow_parser.factory) TensorFlowProfilingParser (class in neural_compressor.profiling.parser.tensorflow_parser.parser) TensorflowQATModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorflowQDQToOnnxQDQConverter (class in neural_compressor.adaptor.tf_utils.tf2onnx_converter) TensorflowQuery (class in neural_compressor.adaptor.tensorflow) (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow) TensorflowRandomHorizontalFlip (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TensorflowRandomVerticalFlip (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TensorflowResizeCropImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform) (class in neural_compressor.experimental.data.transforms.imagenet_transform) TensorflowResizeWithRatio (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform)

TensorflowSavedModelModel (class in neural_compressor.model.tensorflow_model) (class in neural_compressor.tensorflow.utils.model_wrappers) TensorFlowSGD (class in neural_compressor.compression.distillation.optimizers) (class in neural_compressor.experimental.common.optimizer) TensorflowShiftRescale (class in neural_compressor.data.transforms.imagenet_transform) TensorFlowSparseCategoricalCrossentropy (class in neural_compressor.compression.distillation.criterions) TensorflowTFRecordDataset (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowTopK (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) TensorflowTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TensorflowTransforms (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TensorflowTranspose (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TensorflowTransposeLastChannel (class in neural_compressor.data.transforms.imagenet_transform) TensorflowVOCMAP (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) TensorflowVOCRecord (class in neural_compressor.data.datasets.dataset) (class in neural_compressor.experimental.data.datasets.dataset) TensorflowWrapFunction (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) teq_quantize() (in module neural_compressor.adaptor.torch_utils.weight_only) (in module neural_compressor.torch.algorithms.weight_only.teq) TEQLinearFakeQuant (class in neural_compressor.adaptor.torch_utils.model_wrapper) (class in neural_compressor.torch.algorithms.weight_only.modules) TEQuantizer (class in neural_compressor.adaptor.torch_utils.teq) (class in neural_compressor.torch.algorithms.weight_only.teq) TF2ONNXConfig (class in neural_compressor.config) tf_diagnosis_helper() (in module neural_compressor.adaptor.tf_utils.util) tf_to_fp32_onnx() (in module neural_compressor.experimental.export.tf2onnx) tf_to_int8_onnx() (in module neural_compressor.experimental.export.tf2onnx) TFDataDataLoader (class in neural_compressor.data.dataloaders.tensorflow_dataloader) (class in neural_compressor.experimental.data.dataloaders.tensorflow_dataloader) TFModelZooCollectTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TfPruningCallback (class in neural_compressor.experimental.pruning) (class in neural_compressor.experimental.pruning_v2) TFSlimNetsFactory (class in neural_compressor.model.nets_factory) (class in neural_compressor.tensorflow.utils.nets_factory) TFSquadV1ModelZooPostTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TFSquadV1PostTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) TilePattern_1x1 (class in neural_compressor.experimental.pruning_recipes.patterns.tile_pattern) TilePattern_1x16 (class in neural_compressor.experimental.pruning_recipes.patterns.tile_pattern) TilePattern_1x2 (class in neural_compressor.experimental.pruning_recipes.patterns.tile_pattern) TilePattern_2x2 (class in neural_compressor.experimental.pruning_recipes.patterns.tile_pattern) TilePattern_4x1 (class in neural_compressor.experimental.pruning_recipes.patterns.tile_pattern) TilePatternBase (class in neural_compressor.experimental.pruning_recipes.patterns.tile_pattern) time_limit() (in module neural_compressor.utils.utility) to_numpy() (in module neural_compressor.adaptor.ox_utils.util) ToArray (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) ToNDArrayTransform (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) Torch2ONNXConfig (class in neural_compressor.config) torch_to_fp32_onnx() (in module neural_compressor.experimental.export.torch2onnx) torch_to_int8_onnx() (in module neural_compressor.experimental.export.torch2onnx) TorchSmoothQuant (class in neural_compressor.adaptor.torch_utils.waq.smooth_quant) (class in neural_compressor.torch.algorithms.smooth_quant.utility) TpeTuneStrategy (class in neural_compressor.contrib.strategy.tpe) (class in neural_compressor.experimental.contrib.strategy.tpe) trace_and_fuse_sub_graph() (in module neural_compressor.adaptor.torch_utils.symbolic_trace) trace_gptq_target_blocks() (in module neural_compressor.adaptor.torch_utils.gptq) (in module neural_compressor.torch.algorithms.weight_only.gptq) transform_registry() (in module neural_compressor.data.transforms.transform) (in module neural_compressor.experimental.data.transforms.transform) TransformerBasedModelBlockPatternDetector (class in neural_compressor.adaptor.torch_utils.pattern_detector) (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) TRANSFORMS (class in neural_compressor.experimental.data.transforms.transform) Transforms (class in neural_compressor.experimental.data.transforms.transform) Transpose (class in neural_compressor.data.transforms.transform) (class in neural_compressor.experimental.data.transforms.transform) trt_env_setup() (in module neural_compressor.adaptor.ox_utils.util) try_loading_keras() (in module neural_compressor.model.tensorflow_model) (in module neural_compressor.tensorflow.utils.model_wrappers) TuneStrategy (class in neural_compressor.experimental.strategy.strategy) (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.experimental.strategy.utils.tuning_space) (class in neural_compressor.strategy.utils.tuning_space) TuningLogger (class in neural_compressor.common.utils.logger) TuningOrder (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) TuningParam (class in neural_compressor.common.tuning_param) TuningSampler (class in neural_compressor.experimental.strategy.utils.tuning_sampler) (class in neural_compressor.strategy.utils.tuning_sampler) TuningSamplerRegistry (class in neural_compressor.experimental.strategy.utils.tuning_sampler) TuningSpace (class in neural_compressor.experimental.strategy.utils.tuning_space) (class in neural_compressor.strategy.utils.tuning_space)

U

UnaryDirect8BitOperator (class in neural_compressor.adaptor.ox_utils.operators.unary_op) UnaryOperator (class in neural_compressor.adaptor.ox_utils.operators.unary_op) UnicodeRegex (class in neural_compressor.experimental.metric.bleu) (class in neural_compressor.metric.bleu) UnpicklingError, [1] update_frequency_on_step (neural_compressor.experimental.pytorch_pruner.pruner.Pruner attribute)

update_module() (in module neural_compressor.adaptor.torch_utils.layer_wise_quant.utils) (in module neural_compressor.torch.algorithms.layer_wise.utils) update_neural_insights_workload() (in module neural_compressor.utils.neural_insights_utils) update_neural_insights_workload_accuracy_data() (in module neural_compressor.utils.neural_insights_utils) update_params() (in module neural_compressor.compression.pruner.utils) update_sq_scale() (in module neural_compressor.adaptor.torch_utils.util) (in module neural_compressor.torch.algorithms.smooth_quant.utility)

V

valid_reshape_inputs() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (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) ValueInfo (class in neural_compressor.adaptor.ox_utils.util) values_from_const() (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_bn) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_decomposed_in) (in module neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_layer_norm) (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.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_gt_version2() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_gte_version2() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_lt_version2() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility) version1_lte_version2() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.utils.utility) (in module neural_compressor.utils.utility)

W

warn() (in module neural_compressor.utils.logger) warning() (in module neural_compressor.utils.logger) WeightCorrection (class in neural_compressor.algorithm.weight_correction) WeightOnlyQuantSampler (class in neural_compressor.strategy.utils.tuning_sampler) WeightPruningConfig (class in neural_compressor.conf.pythonic_config) (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) (in module neural_compressor.experimental.data.transforms.tokenization)

WordpieceTokenizer (class in neural_compressor.data.transforms.tokenization) (class in neural_compressor.experimental.data.transforms.tokenization) WrapMXNetMetric (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) WrapONNXRTMetric (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) WrappedGPT (class in neural_compressor.compression.pruner.wanda.wrapper) WrapPyTorchMetric (class in neural_compressor.experimental.metric.metric) (class in neural_compressor.metric.metric) write_graph() (in module neural_compressor.adaptor.tf_utils.util) (in module neural_compressor.tensorflow.quantization.utils.utility)

X

XgbSearcher (class in neural_compressor.compression.hpo.search_algorithms)