Intel® Neural Compressor
latest▼
Click link above to switch version
Getting Started
Installation
Examples
APIs
Legal Information
Security Policy
Repo
Intel® Neural Compressor
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
A
AcceleratorRegistry (class in neural_compressor.torch.utils.auto_accelerator)
Accuracy (class in neural_compressor.metric.metric)
(class in neural_compressor.objective)
AccuracyCriterion (class in neural_compressor.config)
acq_max() (in module neural_compressor.strategy.bayesian)
Algorithm (class in neural_compressor.algorithm.algorithm)
algorithm_registry() (in module neural_compressor.algorithm.algorithm)
ALGORITHMS (class in neural_compressor.algorithm.algorithm)
AlgorithmScheduler (class in neural_compressor.algorithm.algorithm)
alias_param() (in module neural_compressor.utils.utility)
AlignImageChannelTransform (class in neural_compressor.data.transforms.transform)
apply_inlining() (in module neural_compressor.tensorflow.quantization.utils.utility)
apply_single_pattern_pair() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
Auto_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)
auto_detect_accelerator() (in module neural_compressor.torch.utils.auto_accelerator)
AutoAlpha (class in neural_compressor.torch.algorithms.smooth_quant.utility)
AutoMixedPrecisionTuneStrategy (class in neural_compressor.strategy.auto_mixed_precision)
autoround_quantize_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
AutoRoundConfig (class in neural_compressor.torch.quantization.config)
AutoRoundQuantizer (class in neural_compressor.torch.algorithms.weight_only.autoround)
autotune() (in module neural_compressor.tensorflow.quantization.autotune)
(in module neural_compressor.torch.quantization.autotune)
AutoTuneStrategy (class in neural_compressor.strategy.auto)
awq_quantize_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
AWQConfig (class in neural_compressor.torch.quantization.config)
AWQQuantizer (class in neural_compressor.torch.algorithms.weight_only.awq)
axis (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)
B
BaseConfig (class in neural_compressor.common.base_config)
BaseDataLoader (class in neural_compressor.tensorflow.utils.data)
BaseMetric (class in neural_compressor.metric.metric)
BaseModel (class in neural_compressor.model.base_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
BaseTransform (class in neural_compressor.data.transforms.transform)
BasicTokenizer (class in neural_compressor.data.transforms.tokenization)
BasicTuneStrategy (class in neural_compressor.strategy.basic)
BatchSampler (class in neural_compressor.tensorflow.utils.data)
BayesianOptimization (class in neural_compressor.strategy.bayesian)
BayesianTuneStrategy (class in neural_compressor.strategy.bayesian)
BenchmarkConfig (class in neural_compressor.config)
BF16Convert (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert)
BiasCorrection (class in neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction)
BilinearImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform)
BLEU (class in neural_compressor.metric.bleu)
bleu_tokenize() (in module neural_compressor.metric.bleu)
block_wise_calibration() (in module neural_compressor.torch.utils.block_wise)
BlockFallbackTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler)
build_captured_dataloader() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
build_slave_faker_model() (in module neural_compressor.strategy.utils.utility)
bypass_reshape() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
C
cal_scale() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
calculate_md5() (in module neural_compressor.data.datasets.dataset)
calculate_mse() (in module neural_compressor.utils.utility)
Calibration (class in neural_compressor.torch.algorithms.smooth_quant.utility)
call_counter() (in module neural_compressor.common.utils.utility)
CallBacks (class in neural_compressor.training)
can_pack_with_numba() (in module neural_compressor.torch.utils.environ)
CaptureOutputToFile (class in neural_compressor.tensorflow.utils.utility)
(class in neural_compressor.utils.utility)
CastONNXTransform (class in neural_compressor.data.transforms.transform)
CastPyTorchTransform (class in neural_compressor.data.transforms.transform)
CastTFTransform (class in neural_compressor.data.transforms.transform)
CenterCropTFTransform (class in neural_compressor.data.transforms.transform)
CenterCropTransform (class in neural_compressor.data.transforms.transform)
cfg_to_qconfig() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.static_quant.utility)
change_config_to_hf_format() (in module neural_compressor.torch.algorithms.weight_only.save_load)
check_cfg_and_qconfig() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.static_quant.utility)
check_integrity() (in module neural_compressor.data.datasets.dataset)
check_key_exist() (in module neural_compressor.utils.utility)
check_model() (in module neural_compressor.utils.export.qlinear2qdq)
checkpoint_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
CIFAR10 (class in neural_compressor.data.datasets.dataset)
CIFAR100 (class in neural_compressor.data.datasets.dataset)
ClassRegister (class in neural_compressor.strategy.utils.utility)
clean_module_weight() (in module neural_compressor.torch.algorithms.layer_wise.utils)
COCOEvalWrapper (class in neural_compressor.metric.coco_tools)
COCOmAPv2 (class in neural_compressor.metric.metric)
COCONpy (class in neural_compressor.data.datasets.coco_dataset)
COCORaw (class in neural_compressor.data.datasets.coco_dataset)
COCORecordDataset (class in neural_compressor.data.datasets.coco_dataset)
COCOWrapper (class in neural_compressor.metric.coco_tools)
collate_tf_preds() (in module neural_compressor.tensorflow.quantization.utils.utility)
CollectTransform (class in neural_compressor.data.transforms.transform)
combine_histogram() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
compare_label (neural_compressor.metric.metric.MAE attribute)
(neural_compressor.metric.metric.MSE attribute)
compare_objects() (in module neural_compressor.utils.utility)
ComposableConfig (class in neural_compressor.common.base_config)
ComposeTransform (class in neural_compressor.data.transforms.transform)
CompressionManager (class in neural_compressor.training)
compute_bleu() (in module neural_compressor.metric.bleu_util)
compute_sparsity() (in module neural_compressor.utils.utility)
config_list (neural_compressor.common.base_config.ComposableConfig attribute)
(neural_compressor.common.base_tuning.ConfigSet attribute)
ConfigLoader (class in neural_compressor.common.base_tuning)
ConfigRegistry (class in neural_compressor.common.base_config)
ConfigSet (class in neural_compressor.common.base_tuning)
ConservativeTuneStrategy (class in neural_compressor.strategy.conservative)
construct_function_from_graph_def() (in module neural_compressor.tensorflow.quantization.utils.utility)
convert() (in module neural_compressor.torch.algorithms.qat.quant_utils)
(in module neural_compressor.torch.quantization.quantize)
convert_by_vocab() (in module neural_compressor.data.transforms.tokenization)
convert_examples_to_features() (in module neural_compressor.data.datasets.bert_dataset)
(in module neural_compressor.data.transforms.transform)
convert_model_with_mapping() (in module neural_compressor.torch.algorithms.qat.quant_utils)
convert_to_unicode() (in module neural_compressor.data.transforms.tokenization)
ConvertAddToBiasAddOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd)
ConvertLayoutOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout)
ConvertLeakyReluOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu)
ConvertNanToRandom (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random)
ConvertPlaceholderToConst (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const)
CPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)
CpuInfo (class in neural_compressor.common.utils.utility)
(class in neural_compressor.tensorflow.utils.utility)
(class in neural_compressor.utils.utility)
create_dataloader() (in module neural_compressor.utils.create_obj_from_config)
create_dataset() (in module neural_compressor.utils.create_obj_from_config)
create_eval_func() (in module neural_compressor.utils.create_obj_from_config)
create_quant_spec_from_config() (in module neural_compressor.torch.algorithms.pt2e_quant.utility)
create_train_func() (in module neural_compressor.utils.create_obj_from_config)
create_xiq_quantizer_from_pt2e_config() (in module neural_compressor.torch.algorithms.pt2e_quant.utility)
CropResizeTFTransform (class in neural_compressor.data.transforms.transform)
CropResizeTransform (class in neural_compressor.data.transforms.transform)
CropToBoundingBox (class in neural_compressor.data.transforms.transform)
CUDA_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)
D
Dataset (class in neural_compressor.data.datasets.dataset)
dataset (neural_compressor.metric.coco_tools.COCOWrapper attribute)
dataset_registry() (in module neural_compressor.data.datasets.dataset)
Datasets (class in neural_compressor.data.datasets.dataset)
debug() (in module neural_compressor.utils.logger)
deep_get() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
deep_set() (in module neural_compressor.utils.utility)
default_collate() (in module neural_compressor.tensorflow.utils.data)
Dequantize() (in module neural_compressor.utils.utility)
dequantize_weight() (in module neural_compressor.utils.utility)
DequantizeCastOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer)
detect_device() (in module neural_compressor.torch.utils.utility)
detect_processor_type_based_on_hw() (in module neural_compressor.common.utils.utility)
detection_type (neural_compressor.metric.coco_tools.COCOWrapper attribute)
device_synchronize() (in module neural_compressor.torch.utils.environ)
DilatedContraction (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction)
disable_random() (in module neural_compressor.tensorflow.utils.utility)
DistillationConfig (class in neural_compressor.config)
DotDict (class in neural_compressor.config)
(class in neural_compressor.utils.utility)
dowload_hf_model() (in module neural_compressor.torch.utils.utility)
download_url() (in module neural_compressor.data.datasets.dataset)
DummyDataset (class in neural_compressor.data.datasets.dummy_dataset)
(class in neural_compressor.data.datasets.dummy_dataset_v2)
(class in neural_compressor.tensorflow.utils.data)
DummyDatasetV2 (class in neural_compressor.tensorflow.utils.data)
dump_class_attrs() (in module neural_compressor.utils.utility)
dump_data_to_local() (in module neural_compressor.utils.utility)
dump_elapsed_time() (in module neural_compressor.common.utils.utility)
(in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
dump_model_op_stats() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.static_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.autoround)
(in module neural_compressor.torch.utils.utility)
dump_table() (in module neural_compressor.utils.utility)
dump_table_to_csv() (in module neural_compressor.utils.utility)
dynamic_quant_export() (in module neural_compressor.utils.export.torch2onnx)
DynamicQuantConfig (class in neural_compressor.torch.quantization.config)
E
ElemFormat (class in neural_compressor.torch.algorithms.mx_quant.utils)
enough_memo_store_scale() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
equal_dicts() (in module neural_compressor.utils.utility)
error() (in module neural_compressor.utils.logger)
estimator_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
evaluate() (in module neural_compressor.metric.evaluate_squad)
(in module neural_compressor.metric.f1)
EvaluationFuncWrapper (class in neural_compressor.common.base_tuning)
Evaluator (class in neural_compressor.common.base_tuning)
exact_match_score() (in module neural_compressor.metric.evaluate_squad)
ExhaustiveTuneStrategy (class in neural_compressor.strategy.exhaustive)
ExpandDimsOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer)
export() (in module neural_compressor.torch.export.pt2e_export)
export_compressed_model() (in module neural_compressor.utils.load_huggingface)
export_hf2compressored_model() (in module neural_compressor.torch.export.export_hf)
export_model_for_pt2e_quant() (in module neural_compressor.torch.export.pt2e_export)
ExportConfig (class in neural_compressor.config)
ExportSingleImageDetectionBoxesToCoco() (in module neural_compressor.metric.coco_tools)
ExportSingleImageDetectionMasksToCoco() (in module neural_compressor.metric.coco_tools)
ExportSingleImageGroundtruthToCoco() (in module neural_compressor.metric.coco_tools)
extract_data_type() (in module neural_compressor.strategy.utils.utility)
F
F1 (class in neural_compressor.metric.metric)
f1_score() (in module neural_compressor.metric.evaluate_squad)
(in module neural_compressor.metric.f1)
FakeAffineTensorQuantFunction (class in neural_compressor.torch.algorithms.weight_only.modules)
FallbackTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler)
FashionMNIST (class in neural_compressor.data.datasets.dataset)
FastBiasCorrection (class in neural_compressor.algorithm.fast_bias_correction)
fatal() (in module neural_compressor.utils.logger)
fault_tolerant_file() (in module neural_compressor.utils.utility)
fetch_module() (in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.utility)
FetchWeightFromReshapeOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape)
Filter (class in neural_compressor.data.filters.filter)
filter_fn() (in module neural_compressor.torch.algorithms.weight_only.hqq.quantizer)
filter_registry() (in module neural_compressor.data.filters.filter)
FILTERS (class in neural_compressor.data.filters.filter)
finalize_calibration() (in module neural_compressor.torch.quantization.quantize)
find_all_layers() (in module neural_compressor.torch.algorithms.weight_only.gptq)
find_layers() (in module neural_compressor.torch.algorithms.weight_only.gptq)
find_layers_name() (in module neural_compressor.torch.algorithms.weight_only.gptq)
find_matching_blocks() (in module neural_compressor.torch.utils.utility)
fit() (in module neural_compressor.mix_precision)
(in module neural_compressor.quantization)
(in module neural_compressor.training)
fix_ref_type_of_graph_def() (in module neural_compressor.tensorflow.quantization.utils.utility)
fn (neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter.PatternPair attribute)
FoldBatchNormNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm)
Footprint (class in neural_compressor.objective)
forward_wrapper() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.utility)
fp8_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
FP8Config (class in neural_compressor.torch.quantization.config)
framework_datasets (in module neural_compressor.data.datasets.dataset)
FreezeFakeQuantOpOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant)
FreezeValueTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value)
frozen_pb_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
FullTokenizer (class in neural_compressor.data.transforms.tokenization)
FuseBiasAddAndAddOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add)
FuseColumnWiseMulOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul)
FuseConvRedundantDequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize)
FuseConvRequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize)
FuseConvWithMathOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math)
FuseDecomposedBNOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
FuseDecomposedINOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
FuseGeluOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu)
FuseLayerNormOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm)
FuseMatMulRedundantDequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize)
FuseMatMulRequantizeDequantizeNewAPITransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize)
FuseMatMulRequantizeDequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize)
FuseMatMulRequantizeNewAPITransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize)
FuseMatMulRequantizeTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize)
FuseNodeStartWithConcatV2 (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_concatv2)
(class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_concatv2)
FuseNodeStartWithConv2d (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_conv)
(class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_conv)
FuseNodeStartWithDeconv2d (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv)
FuseNodeStartWithFusedBatchNormV3 (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_bn)
(class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_bn)
FuseNodeStartWithFusedInstanceNorm (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in)
FuseNodeStartWithMatmul (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_matmul)
(class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_matmul)
FuseNodeStartWithPooling (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_pooling)
(class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling)
FusePadWithConv2DOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv)
FusePadWithFP32Conv2DOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv)
FuseTransposeReshapeOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose)
G
gen_bar_updater() (in module neural_compressor.data.datasets.dataset)
GeneralTopK (class in neural_compressor.metric.metric)
generate_activation_observer() (in module neural_compressor.torch.algorithms.static_quant.utility)
generate_feed_dict() (in module neural_compressor.tensorflow.quantization.utils.utility)
generate_xpu_qconfig() (in module neural_compressor.torch.algorithms.static_quant.utility)
GenerateGraphWithQDQPattern (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern)
get_absorb_layers() (in module neural_compressor.torch.algorithms.weight_only.utility)
get_accelerator() (in module neural_compressor.torch.utils.environ)
get_adaptor_name() (in module neural_compressor.strategy.utils.utility)
get_algorithm() (in module neural_compressor.utils.create_obj_from_config)
get_all_config_set() (in module neural_compressor.tensorflow.quantization.autotune)
(in module neural_compressor.torch.quantization.autotune)
get_all_config_set_from_config_registry() (in module neural_compressor.common.base_config)
get_all_fp32_data() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
get_all_registered_configs() (in module neural_compressor.tensorflow.keras.quantization.config)
(in module neural_compressor.torch.quantization.config)
get_block_names() (in module neural_compressor.torch.utils.utility)
get_block_prefix() (in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.block_wise)
get_children() (in module neural_compressor.torch.algorithms.layer_wise.utils)
get_const_dim_count() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
get_dataloader() (in module neural_compressor.torch.algorithms.weight_only.autoround)
get_default_AutoRound_config() (in module neural_compressor.torch.quantization.config)
get_default_awq_config() (in module neural_compressor.torch.quantization.config)
get_default_double_quant_config() (in module neural_compressor.torch.quantization.config)
get_default_dynamic_config() (in module neural_compressor.torch.quantization.config)
get_default_fp8_config() (in module neural_compressor.torch.quantization.config)
get_default_fp8_config_set() (in module neural_compressor.torch.quantization.config)
get_default_gptq_config() (in module neural_compressor.torch.quantization.config)
get_default_hqq_config() (in module neural_compressor.torch.quantization.config)
get_default_mixed_precision_config() (in module neural_compressor.torch.quantization.config)
get_default_mixed_precision_config_set() (in module neural_compressor.torch.quantization.config)
get_default_mx_config() (in module neural_compressor.torch.quantization.config)
get_default_qat_module_mappings() (in module neural_compressor.torch.quantization.config)
get_default_rtn_config() (in module neural_compressor.torch.quantization.config)
get_default_sq_config() (in module neural_compressor.tensorflow.quantization.config)
(in module neural_compressor.torch.quantization.config)
get_default_static_config() (in module neural_compressor.torch.quantization.config)
get_default_static_quant_config() (in module neural_compressor.tensorflow.keras.quantization.config)
(in module neural_compressor.tensorflow.quantization.config)
get_default_teq_config() (in module neural_compressor.torch.quantization.config)
get_depth() (in module neural_compressor.torch.algorithms.static_quant.utility)
get_dict_at_depth() (in module neural_compressor.torch.algorithms.static_quant.utility)
get_double_quant_config_dict() (in module neural_compressor.torch.utils.utility)
get_element_under_depth() (in module neural_compressor.torch.algorithms.static_quant.utility)
get_enum_from_format() (in module neural_compressor.torch.utils.utility)
get_filter_fn() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
get_final_text() (in module neural_compressor.data.transforms.transform)
get_func_from_config() (in module neural_compressor.utils.create_obj_from_config)
get_graph_def() (in module neural_compressor.tensorflow.quantization.utils.utility)
get_half_precision_node_set() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
get_input_output_node_names() (in module neural_compressor.tensorflow.quantization.utils.utility)
get_ipex_version() (in module neural_compressor.torch.utils.environ)
get_layer_names_in_block() (in module neural_compressor.torch.utils.utility)
get_metrics() (in module neural_compressor.utils.create_obj_from_config)
get_mllm_dataloader() (in module neural_compressor.torch.algorithms.weight_only.autoround)
get_model_device() (in module neural_compressor.torch.utils.utility)
get_model_fwk_name() (in module neural_compressor.model.model)
get_model_info() (in module neural_compressor.torch.utils.utility)
get_model_input_shape() (in module neural_compressor.tensorflow.quantization.utils.utility)
get_model_type() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
get_module() (in module neural_compressor.torch.algorithms.layer_wise.utils)
(in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.utility)
get_module_input_output() (in module neural_compressor.torch.algorithms.weight_only.utility)
get_multimodal_block_names() (in module neural_compressor.torch.utils.utility)
get_named_children() (in module neural_compressor.torch.algorithms.layer_wise.utils)
get_node_mapping() (in module neural_compressor.utils.export.torch2onnx)
get_non_persistent_buffers() (in module neural_compressor.torch.utils.utility)
get_number_of_sockets() (in module neural_compressor.utils.utility)
get_op_list() (in module neural_compressor.utils.utility)
get_parent() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.utility)
get_postprocess() (in module neural_compressor.utils.create_obj_from_config)
get_preprocess() (in module neural_compressor.utils.create_obj_from_config)
get_processor_type_from_user_config() (in module neural_compressor.torch.utils.utility)
get_quant_config() (in module neural_compressor.torch.algorithms.qat.quant_utils)
get_quant_config_with_scheme() (in module neural_compressor.torch.algorithms.qat.quant_utils)
get_quantizable_onnx_ops() (in module neural_compressor.utils.export.torch2onnx)
get_quantizable_ops_from_cfgs() (in module neural_compressor.torch.algorithms.static_quant.utility)
get_quantizable_ops_recursively() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.static_quant.utility)
get_quantization_format() (in module neural_compressor.torch.algorithms.qat.quant_utils)
get_quantizer() (in module neural_compressor.torch.utils.utility)
get_rtn_double_quant_config_set() (in module neural_compressor.torch.quantization.autotune)
get_size() (in module neural_compressor.utils.utility)
get_super_module_by_name() (in module neural_compressor.torch.algorithms.layer_wise.utils)
get_tensor_by_name() (in module neural_compressor.tensorflow.quantization.utils.utility)
get_tensor_histogram() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
get_tensors_info() (in module neural_compressor.utils.utility)
get_tf_model_type() (in module neural_compressor.tensorflow.utils.model_wrappers)
get_torch_version() (in module neural_compressor.torch.utils.environ)
get_torchvision_map() (in module neural_compressor.data.transforms.transform)
get_tuning_history() (in module neural_compressor.utils.utility)
get_unquantized_node_set() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
get_used_cpu_mem_MB() (in module neural_compressor.torch.utils.environ)
get_used_hpu_mem_MB() (in module neural_compressor.torch.utils.environ)
get_weights_details() (in module neural_compressor.utils.utility)
get_woq_tuning_config() (in module neural_compressor.torch.quantization.config)
get_workspace() (in module neural_compressor.common.utils.utility)
GLOBAL_STATE (class in neural_compressor.utils.utility)
GPTQ (class in neural_compressor.torch.algorithms.weight_only.gptq)
gptq_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
GPTQConfig (class in neural_compressor.torch.quantization.config)
GPTQuantizer (class in neural_compressor.torch.algorithms.weight_only.gptq)
graph_def_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
graph_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
GraphAnalyzer (class in neural_compressor.tensorflow.quantization.utils.graph_util)
GraphConverter (class in neural_compressor.tensorflow.quantization.utils.graph_converter)
GraphCseOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer)
GraphFoldConstantOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant)
GraphRewriterBase (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base)
GraphRewriterHelper (class in neural_compressor.tensorflow.quantization.utils.graph_util)
GraphTrace (class in neural_compressor.torch.algorithms.smooth_quant.utility)
(class in neural_compressor.torch.algorithms.weight_only.utility)
GraphTransformBase (class in neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base)
GrapplerOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass)
group_size (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)
H
HalfPrecisionConverter (class in neural_compressor.torch.algorithms.mixed_precision.half_precision_convert)
HalfPrecisionModuleWrapper (class in neural_compressor.torch.algorithms.mixed_precision.module_wrappers)
HAWQ_V2TuneStrategy (class in neural_compressor.strategy.hawq_v2)
HPOConfig (class in neural_compressor.config)
HPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)
HPUWeightOnlyLinear (class in neural_compressor.torch.algorithms.weight_only.modules)
hqq_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
HQQConfig (class in neural_compressor.torch.quantization.config)
HQQLinear (class in neural_compressor.torch.algorithms.weight_only.hqq.core)
HQQModuleConfig (class in neural_compressor.torch.algorithms.weight_only.hqq.config)
HQQTensorHandle (class in neural_compressor.torch.algorithms.weight_only.hqq.core)
HQQuantizer (class in neural_compressor.torch.algorithms.weight_only.hqq.quantizer)
hybrid_gptq_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
HybridGPTQConfig (class in neural_compressor.torch.quantization.config)
I
ImageFolder (class in neural_compressor.data.datasets.dataset)
ImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset)
INCAcceleratorType (class in neural_compressor.torch.utils.auto_accelerator)
INCWeightOnlyLinear (class in neural_compressor.torch.algorithms.weight_only.modules)
IndexFetcher (class in neural_compressor.tensorflow.utils.data)
info() (in module neural_compressor.utils.logger)
init_tuning() (in module neural_compressor.common.base_tuning)
initial_tuning_cfg_with_quant_mode() (in module neural_compressor.strategy.utils.tuning_space)
initialize_int8_avgpool() (in module neural_compressor.tensorflow.keras.layers.pool2d)
initialize_int8_conv2d() (in module neural_compressor.tensorflow.keras.layers.conv2d)
initialize_int8_dense() (in module neural_compressor.tensorflow.keras.layers.dense)
initialize_int8_depthwise_conv2d() (in module neural_compressor.tensorflow.keras.layers.depthwise_conv2d)
initialize_int8_maxpool() (in module neural_compressor.tensorflow.keras.layers.pool2d)
initialize_int8_separable_conv2d() (in module neural_compressor.tensorflow.keras.layers.separable_conv2d)
InjectDummyBiasAddOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd)
InputFeatures (class in neural_compressor.data.datasets.bert_dataset)
(class in neural_compressor.data.transforms.transform)
InsertLogging (class in neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging)
InsertPrintMinMaxNode (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node)
INT8StaticQuantConfig (class in neural_compressor.torch.quantization.config)
IntermediateLayersKnowledgeDistillationLossConfig (class in neural_compressor.config)
IPEXModel (class in neural_compressor.model.torch_model)
is_ckpt_format() (in module neural_compressor.tensorflow.quantization.utils.utility)
is_hpex_available() (in module neural_compressor.torch.utils.environ)
is_hpex_support_g_idx() (in module neural_compressor.torch.utils.environ)
is_hpu_available() (in module neural_compressor.torch.utils.environ)
is_int8_model() (in module neural_compressor.utils.pytorch)
is_ipex_available() (in module neural_compressor.torch.utils.environ)
is_ipex_imported() (in module neural_compressor.torch.utils.environ)
is_leaf() (in module neural_compressor.torch.algorithms.weight_only.gptq)
is_numba_available() (in module neural_compressor.torch.utils.environ)
is_optimum_available() (in module neural_compressor.torch.utils.environ)
is_optimum_habana_available() (in module neural_compressor.torch.utils.environ)
is_package_available() (in module neural_compressor.torch.utils.environ)
is_quantlinear() (in module neural_compressor.torch.algorithms.qat.quant_utils)
is_saved_model_format() (in module neural_compressor.tensorflow.quantization.utils.utility)
is_tbb_available() (in module neural_compressor.torch.utils.environ)
is_transformers_imported() (in module neural_compressor.torch.utils.environ)
IterableDataset (class in neural_compressor.data.datasets.dataset)
IterableFetcher (class in neural_compressor.tensorflow.utils.data)
IterableSampler (class in neural_compressor.tensorflow.utils.data)
iterator_sess_run() (in module neural_compressor.tensorflow.quantization.utils.utility)
itex_installed() (in module neural_compressor.tensorflow.utils.utility)
K
k (neural_compressor.metric.metric.GeneralTopK attribute)
(neural_compressor.metric.metric.TensorflowTopK attribute)
Keras (class in neural_compressor.config)
keras_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
KerasAdaptor (class in neural_compressor.tensorflow.algorithms.static_quant.keras)
KerasConfigConverter (class in neural_compressor.tensorflow.algorithms.static_quant.keras)
KerasModel (class in neural_compressor.model.keras_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
KerasQuery (class in neural_compressor.tensorflow.algorithms.static_quant.keras)
KerasSurgery (class in neural_compressor.tensorflow.algorithms.static_quant.keras)
KL_Divergence (class in neural_compressor.utils.kl_divergence)
KnowledgeDistillationLossConfig (class in neural_compressor.config)
L
label_list (neural_compressor.metric.metric.Accuracy attribute)
(neural_compressor.metric.metric.MAE attribute)
(neural_compressor.metric.metric.MSE attribute)
LabelBalanceCOCORawFilter (class in neural_compressor.data.filters.coco_filter)
LabelBalanceCOCORecordFilter (class in neural_compressor.data.filters.coco_filter)
labels (neural_compressor.metric.bleu.BLEU attribute)
LabelShift (class in neural_compressor.data.transforms.imagenet_transform)
LayerHistogramCollector (class in neural_compressor.utils.collect_layer_histogram)
LazyImport (class in neural_compressor.common.utils.utility)
(class in neural_compressor.utils.utility)
load() (in module neural_compressor.torch.algorithms.layer_wise.load)
(in module neural_compressor.torch.algorithms.pt2e_quant.save_load)
(in module neural_compressor.torch.algorithms.static_quant.save_load)
(in module neural_compressor.torch.algorithms.weight_only.save_load)
(in module neural_compressor.torch.quantization.save_load_entry)
(in module neural_compressor.utils.pytorch)
load_and_cache_examples() (in module neural_compressor.data.datasets.bert_dataset)
load_config_mapping() (in module neural_compressor.common.utils.save_load)
load_data_from_pkl() (in module neural_compressor.utils.utility)
load_empty_model() (in module neural_compressor.torch.utils.utility)
load_first_layer_only() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_layer_wise_quantized_model() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_model_from_shards_with_safetensors() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_module() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_non_persistent_buffers() (in module neural_compressor.torch.utils.utility)
load_saved_model() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
load_tensor() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_tensor_from_safetensors() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_tensor_from_safetensors_shard() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_tensor_from_shard() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_value() (in module neural_compressor.torch.algorithms.layer_wise.utils)
load_vocab() (in module neural_compressor.data.transforms.tokenization)
load_weight_only() (in module neural_compressor.utils.pytorch)
log() (in module neural_compressor.utils.logger)
log_process() (in module neural_compressor.common.utils.utility)
log_quantizable_layers_per_transformer() (in module neural_compressor.torch.algorithms.weight_only.gptq)
Logger (class in neural_compressor.common.utils.logger)
(class in neural_compressor.utils.logger)
Loss (class in neural_compressor.metric.metric)
LowerBitsSampler (class in neural_compressor.strategy.utils.tuning_sampler)
M
MAE (class in neural_compressor.metric.metric)
Matmul (class in neural_compressor.torch.algorithms.weight_only.modules)
MergeDuplicatedQDQOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq)
MetaInfoChangingMemOpOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer)
Metric (class in neural_compressor.metric.metric)
metric_max_over_ground_truths() (in module neural_compressor.metric.evaluate_squad)
(in module neural_compressor.metric.f1)
metric_registry() (in module neural_compressor.metric.metric)
METRICS (class in neural_compressor.metric.metric)
metrics (neural_compressor.metric.metric.METRICS attribute)
(neural_compressor.metric.metric.ONNXRTITMetrics attribute)
(neural_compressor.metric.metric.ONNXRTQLMetrics attribute)
(neural_compressor.metric.metric.PyTorchMetrics attribute)
(neural_compressor.metric.metric.TensorflowMetrics attribute)
mIOU (class in neural_compressor.metric.metric)
mixed_precision_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
MixedPrecisionConfig (class in neural_compressor.config)
(class in neural_compressor.torch.quantization.config)
MNIST (class in neural_compressor.data.datasets.dataset)
Mode (class in neural_compressor.common.utils.constants)
MODE (class in neural_compressor.utils.utility)
Model (class in neural_compressor.model.model)
(class in neural_compressor.tensorflow.utils.model)
model_forward() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.utility)
model_forward_per_sample() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
MODEL_LEVEL (neural_compressor.common.tuning_param.ParamLevel attribute)
model_type (neural_compressor.model.tensorflow_model.TensorflowQATModel attribute)
ModelSize (class in neural_compressor.objective)
ModelWiseTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler)
module
neural_compressor
neural_compressor.algorithm
neural_compressor.algorithm.algorithm
neural_compressor.algorithm.fast_bias_correction
neural_compressor.algorithm.smooth_quant
neural_compressor.algorithm.weight_correction
neural_compressor.common
neural_compressor.common.base_config
neural_compressor.common.base_tuning
neural_compressor.common.tuning_param
neural_compressor.common.utils
neural_compressor.common.utils.constants
neural_compressor.common.utils.logger
neural_compressor.common.utils.save_load
neural_compressor.common.utils.utility
neural_compressor.common.version
neural_compressor.config
neural_compressor.contrib
neural_compressor.contrib.strategy
neural_compressor.contrib.strategy.tpe
neural_compressor.data
neural_compressor.data.datasets
neural_compressor.data.datasets.bert_dataset
neural_compressor.data.datasets.coco_dataset
neural_compressor.data.datasets.dataset
neural_compressor.data.datasets.dummy_dataset
neural_compressor.data.datasets.dummy_dataset_v2
neural_compressor.data.datasets.imagenet_dataset
neural_compressor.data.datasets.style_transfer_dataset
neural_compressor.data.filters
neural_compressor.data.filters.coco_filter
neural_compressor.data.filters.filter
neural_compressor.data.transforms
neural_compressor.data.transforms.imagenet_transform
neural_compressor.data.transforms.postprocess
neural_compressor.data.transforms.tokenization
neural_compressor.data.transforms.transform
neural_compressor.metric
neural_compressor.metric.bleu
neural_compressor.metric.bleu_util
neural_compressor.metric.coco_label_map
neural_compressor.metric.coco_tools
neural_compressor.metric.evaluate_squad
neural_compressor.metric.f1
neural_compressor.metric.metric
neural_compressor.mix_precision
neural_compressor.model
neural_compressor.model.base_model
neural_compressor.model.keras_model
neural_compressor.model.model
neural_compressor.model.nets_factory
neural_compressor.model.onnx_model
neural_compressor.model.tensorflow_model
neural_compressor.model.torch_model
neural_compressor.objective
neural_compressor.profiling
neural_compressor.quantization
neural_compressor.strategy
neural_compressor.strategy.auto
neural_compressor.strategy.auto_mixed_precision
neural_compressor.strategy.basic
neural_compressor.strategy.bayesian
neural_compressor.strategy.conservative
neural_compressor.strategy.exhaustive
neural_compressor.strategy.hawq_v2
neural_compressor.strategy.mse
neural_compressor.strategy.mse_v2
neural_compressor.strategy.random
neural_compressor.strategy.strategy
neural_compressor.strategy.utils
neural_compressor.strategy.utils.constant
neural_compressor.strategy.utils.tuning_sampler
neural_compressor.strategy.utils.tuning_space
neural_compressor.strategy.utils.tuning_structs
neural_compressor.strategy.utils.utility
neural_compressor.tensorflow
neural_compressor.tensorflow.algorithms
neural_compressor.tensorflow.algorithms.smoother
neural_compressor.tensorflow.algorithms.smoother.calibration
neural_compressor.tensorflow.algorithms.smoother.core
neural_compressor.tensorflow.algorithms.smoother.scaler
neural_compressor.tensorflow.algorithms.static_quant
neural_compressor.tensorflow.algorithms.static_quant.keras
neural_compressor.tensorflow.algorithms.static_quant.tensorflow
neural_compressor.tensorflow.keras
neural_compressor.tensorflow.keras.layers
neural_compressor.tensorflow.keras.layers.conv2d
neural_compressor.tensorflow.keras.layers.dense
neural_compressor.tensorflow.keras.layers.depthwise_conv2d
neural_compressor.tensorflow.keras.layers.layer_initializer
neural_compressor.tensorflow.keras.layers.pool2d
neural_compressor.tensorflow.keras.layers.separable_conv2d
neural_compressor.tensorflow.keras.quantization
neural_compressor.tensorflow.keras.quantization.config
neural_compressor.tensorflow.quantization
neural_compressor.tensorflow.quantization.algorithm_entry
neural_compressor.tensorflow.quantization.autotune
neural_compressor.tensorflow.quantization.config
neural_compressor.tensorflow.quantization.quantize
neural_compressor.tensorflow.quantization.utils
neural_compressor.tensorflow.quantization.utils.graph_converter
neural_compressor.tensorflow.quantization.utils.graph_rewriter
neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16
neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert
neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer
neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern
neural_compressor.tensorflow.quantization.utils.graph_util
neural_compressor.tensorflow.quantization.utils.quantize_graph
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_bn
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_concatv2
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_conv
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_matmul
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_pooling
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_bn
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_concatv2
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_conv
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_matmul
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling
neural_compressor.tensorflow.quantization.utils.quantize_graph_common
neural_compressor.tensorflow.quantization.utils.transform_graph
neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction
neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base
neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging
neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat
neural_compressor.tensorflow.quantization.utils.utility
neural_compressor.tensorflow.utils
neural_compressor.tensorflow.utils.constants
neural_compressor.tensorflow.utils.data
neural_compressor.tensorflow.utils.model
neural_compressor.tensorflow.utils.model_wrappers
neural_compressor.tensorflow.utils.utility
neural_compressor.torch
neural_compressor.torch.algorithms
neural_compressor.torch.algorithms.base_algorithm
neural_compressor.torch.algorithms.layer_wise
neural_compressor.torch.algorithms.layer_wise.load
neural_compressor.torch.algorithms.layer_wise.modified_pickle
neural_compressor.torch.algorithms.layer_wise.utils
neural_compressor.torch.algorithms.mixed_precision
neural_compressor.torch.algorithms.mixed_precision.half_precision_convert
neural_compressor.torch.algorithms.mixed_precision.module_wrappers
neural_compressor.torch.algorithms.mx_quant
neural_compressor.torch.algorithms.mx_quant.mx
neural_compressor.torch.algorithms.mx_quant.utils
neural_compressor.torch.algorithms.pt2e_quant
neural_compressor.torch.algorithms.pt2e_quant.core
neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter
neural_compressor.torch.algorithms.pt2e_quant.save_load
neural_compressor.torch.algorithms.pt2e_quant.utility
neural_compressor.torch.algorithms.qat
neural_compressor.torch.algorithms.qat.quant_linear
neural_compressor.torch.algorithms.qat.quant_utils
neural_compressor.torch.algorithms.qat.tensor_quantizer
neural_compressor.torch.algorithms.smooth_quant
neural_compressor.torch.algorithms.smooth_quant.save_load
neural_compressor.torch.algorithms.smooth_quant.smooth_quant
neural_compressor.torch.algorithms.smooth_quant.utility
neural_compressor.torch.algorithms.static_quant
neural_compressor.torch.algorithms.static_quant.save_load
neural_compressor.torch.algorithms.static_quant.static_quant
neural_compressor.torch.algorithms.static_quant.utility
neural_compressor.torch.algorithms.weight_only
neural_compressor.torch.algorithms.weight_only.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.quantizer
neural_compressor.torch.algorithms.weight_only.modules
neural_compressor.torch.algorithms.weight_only.rtn
neural_compressor.torch.algorithms.weight_only.save_load
neural_compressor.torch.algorithms.weight_only.teq
neural_compressor.torch.algorithms.weight_only.utility
neural_compressor.torch.export
neural_compressor.torch.export.export_hf
neural_compressor.torch.export.pt2e_export
neural_compressor.torch.quantization
neural_compressor.torch.quantization.algorithm_entry
neural_compressor.torch.quantization.autotune
neural_compressor.torch.quantization.config
neural_compressor.torch.quantization.quantize
neural_compressor.torch.quantization.save_load_entry
neural_compressor.torch.utils
neural_compressor.torch.utils.auto_accelerator
neural_compressor.torch.utils.bit_packer
neural_compressor.torch.utils.block_wise
neural_compressor.torch.utils.constants
neural_compressor.torch.utils.environ
neural_compressor.torch.utils.utility
neural_compressor.training
neural_compressor.utils
neural_compressor.utils.collect_layer_histogram
neural_compressor.utils.constant
neural_compressor.utils.create_obj_from_config
neural_compressor.utils.export
neural_compressor.utils.export.qlinear2qdq
neural_compressor.utils.export.tf2onnx
neural_compressor.utils.export.torch2onnx
neural_compressor.utils.kl_divergence
neural_compressor.utils.load_huggingface
neural_compressor.utils.logger
neural_compressor.utils.options
neural_compressor.utils.pytorch
neural_compressor.utils.utility
neural_compressor.utils.weights_details
neural_compressor.version
move_input_device() (in module neural_compressor.torch.utils.utility)
move_input_to_device() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.utility)
MoveSqueezeAfterReluOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu)
MSE (class in neural_compressor.metric.metric)
mse (neural_compressor.metric.metric.RMSE attribute)
mse_metric_gap() (in module neural_compressor.utils.utility)
MSE_V2TuneStrategy (class in neural_compressor.strategy.mse_v2)
MSETuneStrategy (class in neural_compressor.strategy.mse)
MulLinear (class in neural_compressor.torch.algorithms.weight_only.modules)
MultiObjective (class in neural_compressor.objective)
mx_quant_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
MXLinear (class in neural_compressor.torch.algorithms.mx_quant.mx)
MXQuantConfig (class in neural_compressor.torch.quantization.config)
MXQuantizer (class in neural_compressor.torch.algorithms.mx_quant.mx)
N
name (neural_compressor.common.base_config.BaseConfig attribute)
NASConfig (class in neural_compressor.config)
nbits (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)
need_apply() (in module neural_compressor.tensorflow.quantization.quantize)
(in module neural_compressor.torch.quantization.quantize)
neural_compressor
module
neural_compressor.algorithm
module
neural_compressor.algorithm.algorithm
module
neural_compressor.algorithm.fast_bias_correction
module
neural_compressor.algorithm.smooth_quant
module
neural_compressor.algorithm.weight_correction
module
neural_compressor.common
module
neural_compressor.common.base_config
module
neural_compressor.common.base_tuning
module
neural_compressor.common.tuning_param
module
neural_compressor.common.utils
module
neural_compressor.common.utils.constants
module
neural_compressor.common.utils.logger
module
neural_compressor.common.utils.save_load
module
neural_compressor.common.utils.utility
module
neural_compressor.common.version
module
neural_compressor.config
module
neural_compressor.contrib
module
neural_compressor.contrib.strategy
module
neural_compressor.contrib.strategy.tpe
module
neural_compressor.data
module
neural_compressor.data.datasets
module
neural_compressor.data.datasets.bert_dataset
module
neural_compressor.data.datasets.coco_dataset
module
neural_compressor.data.datasets.dataset
module
neural_compressor.data.datasets.dummy_dataset
module
neural_compressor.data.datasets.dummy_dataset_v2
module
neural_compressor.data.datasets.imagenet_dataset
module
neural_compressor.data.datasets.style_transfer_dataset
module
neural_compressor.data.filters
module
neural_compressor.data.filters.coco_filter
module
neural_compressor.data.filters.filter
module
neural_compressor.data.transforms
module
neural_compressor.data.transforms.imagenet_transform
module
neural_compressor.data.transforms.postprocess
module
neural_compressor.data.transforms.tokenization
module
neural_compressor.data.transforms.transform
module
neural_compressor.metric
module
neural_compressor.metric.bleu
module
neural_compressor.metric.bleu_util
module
neural_compressor.metric.coco_label_map
module
neural_compressor.metric.coco_tools
module
neural_compressor.metric.evaluate_squad
module
neural_compressor.metric.f1
module
neural_compressor.metric.metric
module
neural_compressor.mix_precision
module
neural_compressor.model
module
neural_compressor.model.base_model
module
neural_compressor.model.keras_model
module
neural_compressor.model.model
module
neural_compressor.model.nets_factory
module
neural_compressor.model.onnx_model
module
neural_compressor.model.tensorflow_model
module
neural_compressor.model.torch_model
module
neural_compressor.objective
module
neural_compressor.profiling
module
neural_compressor.quantization
module
neural_compressor.strategy
module
neural_compressor.strategy.auto
module
neural_compressor.strategy.auto_mixed_precision
module
neural_compressor.strategy.basic
module
neural_compressor.strategy.bayesian
module
neural_compressor.strategy.conservative
module
neural_compressor.strategy.exhaustive
module
neural_compressor.strategy.hawq_v2
module
neural_compressor.strategy.mse
module
neural_compressor.strategy.mse_v2
module
neural_compressor.strategy.random
module
neural_compressor.strategy.strategy
module
neural_compressor.strategy.utils
module
neural_compressor.strategy.utils.constant
module
neural_compressor.strategy.utils.tuning_sampler
module
neural_compressor.strategy.utils.tuning_space
module
neural_compressor.strategy.utils.tuning_structs
module
neural_compressor.strategy.utils.utility
module
neural_compressor.tensorflow
module
neural_compressor.tensorflow.algorithms
module
neural_compressor.tensorflow.algorithms.smoother
module
neural_compressor.tensorflow.algorithms.smoother.calibration
module
neural_compressor.tensorflow.algorithms.smoother.core
module
neural_compressor.tensorflow.algorithms.smoother.scaler
module
neural_compressor.tensorflow.algorithms.static_quant
module
neural_compressor.tensorflow.algorithms.static_quant.keras
module
neural_compressor.tensorflow.algorithms.static_quant.tensorflow
module
neural_compressor.tensorflow.keras
module
neural_compressor.tensorflow.keras.layers
module
neural_compressor.tensorflow.keras.layers.conv2d
module
neural_compressor.tensorflow.keras.layers.dense
module
neural_compressor.tensorflow.keras.layers.depthwise_conv2d
module
neural_compressor.tensorflow.keras.layers.layer_initializer
module
neural_compressor.tensorflow.keras.layers.pool2d
module
neural_compressor.tensorflow.keras.layers.separable_conv2d
module
neural_compressor.tensorflow.keras.quantization
module
neural_compressor.tensorflow.keras.quantization.config
module
neural_compressor.tensorflow.quantization
module
neural_compressor.tensorflow.quantization.algorithm_entry
module
neural_compressor.tensorflow.quantization.autotune
module
neural_compressor.tensorflow.quantization.config
module
neural_compressor.tensorflow.quantization.quantize
module
neural_compressor.tensorflow.quantization.utils
module
neural_compressor.tensorflow.quantization.utils.graph_converter
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq
module
neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern
module
neural_compressor.tensorflow.quantization.utils.graph_util
module
neural_compressor.tensorflow.quantization.utils.quantize_graph
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_bn
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_concatv2
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_conv
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_deconv
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_in
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_matmul
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.fuse_qdq_pooling
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_bn
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_concatv2
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_conv
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_matmul
module
neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling
module
neural_compressor.tensorflow.quantization.utils.quantize_graph_common
module
neural_compressor.tensorflow.quantization.utils.transform_graph
module
neural_compressor.tensorflow.quantization.utils.transform_graph.bias_correction
module
neural_compressor.tensorflow.quantization.utils.transform_graph.graph_transform_base
module
neural_compressor.tensorflow.quantization.utils.transform_graph.insert_logging
module
neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat
module
neural_compressor.tensorflow.quantization.utils.utility
module
neural_compressor.tensorflow.utils
module
neural_compressor.tensorflow.utils.constants
module
neural_compressor.tensorflow.utils.data
module
neural_compressor.tensorflow.utils.model
module
neural_compressor.tensorflow.utils.model_wrappers
module
neural_compressor.tensorflow.utils.utility
module
neural_compressor.torch
module
neural_compressor.torch.algorithms
module
neural_compressor.torch.algorithms.base_algorithm
module
neural_compressor.torch.algorithms.layer_wise
module
neural_compressor.torch.algorithms.layer_wise.load
module
neural_compressor.torch.algorithms.layer_wise.modified_pickle
module
neural_compressor.torch.algorithms.layer_wise.utils
module
neural_compressor.torch.algorithms.mixed_precision
module
neural_compressor.torch.algorithms.mixed_precision.half_precision_convert
module
neural_compressor.torch.algorithms.mixed_precision.module_wrappers
module
neural_compressor.torch.algorithms.mx_quant
module
neural_compressor.torch.algorithms.mx_quant.mx
module
neural_compressor.torch.algorithms.mx_quant.utils
module
neural_compressor.torch.algorithms.pt2e_quant
module
neural_compressor.torch.algorithms.pt2e_quant.core
module
neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter
module
neural_compressor.torch.algorithms.pt2e_quant.save_load
module
neural_compressor.torch.algorithms.pt2e_quant.utility
module
neural_compressor.torch.algorithms.qat
module
neural_compressor.torch.algorithms.qat.quant_linear
module
neural_compressor.torch.algorithms.qat.quant_utils
module
neural_compressor.torch.algorithms.qat.tensor_quantizer
module
neural_compressor.torch.algorithms.smooth_quant
module
neural_compressor.torch.algorithms.smooth_quant.save_load
module
neural_compressor.torch.algorithms.smooth_quant.smooth_quant
module
neural_compressor.torch.algorithms.smooth_quant.utility
module
neural_compressor.torch.algorithms.static_quant
module
neural_compressor.torch.algorithms.static_quant.save_load
module
neural_compressor.torch.algorithms.static_quant.static_quant
module
neural_compressor.torch.algorithms.static_quant.utility
module
neural_compressor.torch.algorithms.weight_only
module
neural_compressor.torch.algorithms.weight_only.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.quantizer
module
neural_compressor.torch.algorithms.weight_only.modules
module
neural_compressor.torch.algorithms.weight_only.rtn
module
neural_compressor.torch.algorithms.weight_only.save_load
module
neural_compressor.torch.algorithms.weight_only.teq
module
neural_compressor.torch.algorithms.weight_only.utility
module
neural_compressor.torch.export
module
neural_compressor.torch.export.export_hf
module
neural_compressor.torch.export.pt2e_export
module
neural_compressor.torch.quantization
module
neural_compressor.torch.quantization.algorithm_entry
module
neural_compressor.torch.quantization.autotune
module
neural_compressor.torch.quantization.config
module
neural_compressor.torch.quantization.quantize
module
neural_compressor.torch.quantization.save_load_entry
module
neural_compressor.torch.utils
module
neural_compressor.torch.utils.auto_accelerator
module
neural_compressor.torch.utils.bit_packer
module
neural_compressor.torch.utils.block_wise
module
neural_compressor.torch.utils.constants
module
neural_compressor.torch.utils.environ
module
neural_compressor.torch.utils.utility
module
neural_compressor.training
module
neural_compressor.utils
module
neural_compressor.utils.collect_layer_histogram
module
neural_compressor.utils.constant
module
neural_compressor.utils.create_obj_from_config
module
neural_compressor.utils.export
module
neural_compressor.utils.export.qlinear2qdq
module
neural_compressor.utils.export.tf2onnx
module
neural_compressor.utils.export.torch2onnx
module
neural_compressor.utils.kl_divergence
module
neural_compressor.utils.load_huggingface
module
neural_compressor.utils.logger
module
neural_compressor.utils.options
module
neural_compressor.utils.pytorch
module
neural_compressor.utils.utility
module
neural_compressor.utils.weights_details
module
neural_compressor.version
module
node_from_map() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm)
node_name_from_input() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm)
nondigit_punct_re (neural_compressor.metric.bleu.UnicodeRegex attribute)
normalize_answer() (in module neural_compressor.metric.f1)
NormalizeTFTransform (class in neural_compressor.data.transforms.transform)
NormalizeTransform (class in neural_compressor.data.transforms.transform)
num_correct (neural_compressor.metric.metric.GeneralTopK attribute)
(neural_compressor.metric.metric.TensorflowTopK attribute)
num_sample (neural_compressor.metric.metric.GeneralTopK attribute)
(neural_compressor.metric.metric.TensorflowTopK attribute)
O
Objective (class in neural_compressor.objective)
objective_custom_registry() (in module neural_compressor.objective)
objective_registry() (in module neural_compressor.objective)
ONNX (class in neural_compressor.config)
onnx_qlinear_to_qdq() (in module neural_compressor.utils.export.qlinear2qdq)
OnnxBilinearImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform)
ONNXModel (class in neural_compressor.model.onnx_model)
ONNXQlinear2QDQConfig (class in neural_compressor.config)
ONNXResizeCropImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform)
onnxrt (class in neural_compressor.utils.options)
ONNXRTBertDataset (class in neural_compressor.data.datasets.bert_dataset)
ONNXRTCropToBoundingBox (class in neural_compressor.data.transforms.transform)
ONNXRTGLUE (class in neural_compressor.metric.metric)
ONNXRTImagenetDataset (class in neural_compressor.data.datasets.imagenet_dataset)
ONNXRTITDatasets (class in neural_compressor.data.datasets.dataset)
ONNXRTITFilters (class in neural_compressor.data.filters.filter)
ONNXRTITMetrics (class in neural_compressor.metric.metric)
ONNXRTITTransforms (class in neural_compressor.data.transforms.transform)
ONNXRTQLDatasets (class in neural_compressor.data.datasets.dataset)
ONNXRTQLFilters (class in neural_compressor.data.filters.filter)
ONNXRTQLMetrics (class in neural_compressor.metric.metric)
ONNXRTQLTransforms (class in neural_compressor.data.transforms.transform)
OP_LEVEL (neural_compressor.common.tuning_param.ParamLevel attribute)
OP_TYPE_LEVEL (neural_compressor.common.tuning_param.ParamLevel attribute)
OpEntry (class in neural_compressor.utils.utility)
OperatorConfig (class in neural_compressor.tensorflow.keras.quantization.config)
(class in neural_compressor.torch.quantization.config)
optimize_weights_proximal_legacy() (in module neural_compressor.torch.algorithms.weight_only.hqq.optimizer)
OptimizedModel (class in neural_compressor.utils.load_huggingface)
OptimizeQDQGraph (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.qdq.optimize_qdq)
Options (class in neural_compressor.config)
OpTuningConfig (class in neural_compressor.strategy.utils.tuning_structs)
OpTypeWiseTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler)
OpWiseTuningSampler (class in neural_compressor.strategy.utils.tuning_sampler)
OrderedDefaultDict (class in neural_compressor.strategy.utils.utility)
P
pack_array_with_numba_b2_c16() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b2_c32() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b2_c64() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b2_c8() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b4_c16() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b4_c32() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b4_c64() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b4_c8() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b8_c16() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b8_c32() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b8_c64() (in module neural_compressor.torch.utils.bit_packer)
pack_array_with_numba_b8_c8() (in module neural_compressor.torch.utils.bit_packer)
Packer (class in neural_compressor.torch.algorithms.weight_only.hqq.bitpack)
packing (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)
PaddedCenterCropTransform (class in neural_compressor.data.transforms.transform)
ParamLevel (class in neural_compressor.common.tuning_param)
params_list (neural_compressor.common.base_config.BaseConfig attribute)
parse_cfgs() (in module neural_compressor.torch.algorithms.static_quant.utility)
parse_saved_model() (in module neural_compressor.tensorflow.quantization.utils.utility)
ParseDecodeBert (class in neural_compressor.data.datasets.bert_dataset)
ParseDecodeCoco (class in neural_compressor.data.datasets.coco_dataset)
ParseDecodeImagenet (class in neural_compressor.data.transforms.imagenet_transform)
ParseDecodeImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform)
ParseDecodeVocTransform (class in neural_compressor.data.transforms.transform)
patch_hqq_moduile() (in module neural_compressor.torch.algorithms.weight_only.hqq.quantizer)
pattern_factory() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
pattern_to_internal() (in module neural_compressor.strategy.utils.tuning_space)
pattern_to_path() (in module neural_compressor.strategy.utils.tuning_space)
PatternPair (class in neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
Performance (class in neural_compressor.objective)
PickleError
PicklingError
PostCseOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse)
PostHostConstConverter (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter)
Postprocess (class in neural_compressor.data.transforms.postprocess)
postprocess_model() (in module neural_compressor.torch.utils.utility)
PostTrainingQuantConfig (class in neural_compressor.config)
pred_list (neural_compressor.metric.metric.Accuracy attribute)
(neural_compressor.metric.metric.MAE attribute)
(neural_compressor.metric.metric.MSE attribute)
predictions (neural_compressor.metric.bleu.BLEU attribute)
PreOptimization (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize)
prepare() (in module neural_compressor.torch.quantization.quantize)
prepare_compression() (in module neural_compressor.training)
prepare_qat() (in module neural_compressor.torch.quantization.quantize)
preprocess_quant_config() (in module neural_compressor.torch.quantization.quantize)
preprocess_user_cfg() (in module neural_compressor.strategy.utils.utility)
print_op_list() (in module neural_compressor.utils.utility)
print_table() (in module neural_compressor.utils.utility)
ProcessorType (class in neural_compressor.common.utils.utility)
pt2e_dynamic_quant_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
pt2e_static_quant_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
punct_nondigit_re (neural_compressor.metric.bleu.UnicodeRegex attribute)
PyTorch (class in neural_compressor.config)
PyTorchAlignImageChannel (class in neural_compressor.data.transforms.transform)
PyTorchBaseModel (class in neural_compressor.model.torch_model)
PytorchBertDataset (class in neural_compressor.data.datasets.bert_dataset)
PytorchCIFAR10 (class in neural_compressor.data.datasets.dataset)
PytorchCIFAR100 (class in neural_compressor.data.datasets.dataset)
PyTorchCropResizeTransform (class in neural_compressor.data.transforms.transform)
PyTorchDatasets (class in neural_compressor.data.datasets.dataset)
PytorchFashionMNIST (class in neural_compressor.data.datasets.dataset)
PyTorchFilters (class in neural_compressor.data.filters.filter)
PyTorchFXModel (class in neural_compressor.model.torch_model)
PytorchImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset)
PyTorchLoss (class in neural_compressor.metric.metric)
PyTorchMetrics (class in neural_compressor.metric.metric)
PytorchMNIST (class in neural_compressor.data.datasets.dataset)
PyTorchModel (class in neural_compressor.model.torch_model)
PyTorchNormalizeTransform (class in neural_compressor.data.transforms.transform)
PytorchTransform (class in neural_compressor.data.transforms.transform)
PyTorchTransforms (class in neural_compressor.data.transforms.transform)
PyTorchTranspose (class in neural_compressor.data.transforms.transform)
PytorchWrapDataset (class in neural_compressor.data.datasets.dataset)
PytorchWrapFunction (class in neural_compressor.data.datasets.dataset)
(class in neural_compressor.data.transforms.transform)
Q
QAvgPool2D (class in neural_compressor.tensorflow.keras.layers.pool2d)
QConv2D (class in neural_compressor.tensorflow.keras.layers.conv2d)
QDense (class in neural_compressor.tensorflow.keras.layers.dense)
QDepthwiseConv2D (class in neural_compressor.tensorflow.keras.layers.depthwise_conv2d)
qdq_quantize() (in module neural_compressor.torch.algorithms.smooth_quant.smooth_quant)
qdq_weight_actor() (in module neural_compressor.torch.algorithms.weight_only.utility)
qdq_weight_asym() (in module neural_compressor.torch.algorithms.weight_only.utility)
qdq_weight_sym() (in module neural_compressor.torch.algorithms.weight_only.utility)
QDQLayer (class in neural_compressor.torch.algorithms.layer_wise.utils)
(class in neural_compressor.torch.algorithms.weight_only.modules)
QMaxPool2D (class in neural_compressor.tensorflow.keras.layers.pool2d)
QSeparableConv2D (class in neural_compressor.tensorflow.keras.layers.separable_conv2d)
QTensor (class in neural_compressor.torch.algorithms.weight_only.hqq.qtensor)
QTensorConfig (class in neural_compressor.torch.algorithms.weight_only.hqq.config)
QTensorMetaInfo (class in neural_compressor.torch.algorithms.weight_only.hqq.qtensor)
quant_dequant_w_v1() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
quant_dequant_x_v1() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
quant_mode_from_pattern() (in module neural_compressor.strategy.utils.tuning_space)
quant_tensor() (in module neural_compressor.torch.algorithms.weight_only.utility)
quant_weight_w_scale() (in module neural_compressor.torch.algorithms.weight_only.utility)
QuantizationAwareTrainingConfig (class in neural_compressor.config)
quantize() (in module neural_compressor.torch.quantization.quantize)
quantize_4bit() (in module neural_compressor.torch.algorithms.weight_only.utility)
quantize_elemwise_op() (in module neural_compressor.torch.algorithms.mx_quant.utils)
quantize_model() (in module neural_compressor.tensorflow.quantization.quantize)
quantize_model_with_single_config() (in module neural_compressor.tensorflow.quantization.quantize)
quantize_mx_op() (in module neural_compressor.torch.algorithms.mx_quant.utils)
QuantizedInput (class in neural_compressor.data.transforms.imagenet_transform)
QuantizeGraphBase (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base)
QuantizeGraphForIntel (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_for_intel_cpu)
QuantizeGraphHelper (class in neural_compressor.tensorflow.quantization.utils.quantize_graph_common)
QuantizeNodeBase (class in neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base)
Quantizer (class in neural_compressor.torch.algorithms.base_algorithm)
(class in neural_compressor.torch.algorithms.weight_only.gptq)
QuantLinear (class in neural_compressor.torch.algorithms.qat.quant_linear)
QuantOptions (class in neural_compressor.strategy.utils.utility)
QuantType (class in neural_compressor.strategy.utils.utility)
R
RandomCropTFTransform (class in neural_compressor.data.transforms.transform)
RandomCropTransform (class in neural_compressor.data.transforms.transform)
RandomHorizontalFlip (class in neural_compressor.data.transforms.transform)
RandomResizedCropPytorchTransform (class in neural_compressor.data.transforms.transform)
RandomResizedCropTFTransform (class in neural_compressor.data.transforms.transform)
RandomResizedCropTransform (class in neural_compressor.data.transforms.transform)
RandomTuneStrategy (class in neural_compressor.strategy.random)
RandomVerticalFlip (class in neural_compressor.data.transforms.transform)
RAWGPTQuantizer (class in neural_compressor.torch.algorithms.weight_only.gptq)
read_graph() (in module neural_compressor.tensorflow.quantization.utils.utility)
read_json_file() (in module neural_compressor.torch.utils.utility)
read_squad_examples() (in module neural_compressor.data.transforms.transform)
reconstruct_saved_model() (in module neural_compressor.tensorflow.quantization.utils.utility)
recover() (in module neural_compressor.utils.utility)
recover_forward() (in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.block_wise)
recover_model_from_json() (in module neural_compressor.torch.algorithms.smooth_quant.save_load)
(in module neural_compressor.utils.pytorch)
register_accelerator() (in module neural_compressor.torch.utils.auto_accelerator)
register_algo() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.torch.utils.utility)
register_autotune() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
register_config() (in module neural_compressor.common.base_config)
register_customer_metric() (in module neural_compressor.metric.metric)
register_pack_func() (in module neural_compressor.torch.utils.bit_packer)
register_supported_configs_for_fwk() (in module neural_compressor.common.base_config)
register_weight_hooks() (in module neural_compressor.torch.algorithms.layer_wise.utils)
RemoveTrainingNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes)
RenameBatchNormOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm)
replace_forward() (in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.block_wise)
replace_pattern (neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter.PatternPair attribute)
replace_with_quant_linear() (in module neural_compressor.torch.algorithms.qat.quant_utils)
replacement_fn() (in module neural_compressor.torch.algorithms.weight_only.hqq.quantizer)
RerangeQuantizedConcat (class in neural_compressor.tensorflow.quantization.utils.transform_graph.rerange_quantized_concat)
RescaleKerasPretrainTransform (class in neural_compressor.data.transforms.transform)
RescaleTFTransform (class in neural_compressor.data.transforms.transform)
RescaleTransform (class in neural_compressor.data.transforms.transform)
reshape_in_channel_to_last() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
reshape_scale_as_input() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
reshape_scale_as_weight() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
ResizePytorchTransform (class in neural_compressor.data.transforms.transform)
ResizeTFTransform (class in neural_compressor.data.transforms.transform)
ResizeTransform (class in neural_compressor.data.transforms.transform)
ResizeWithAspectRatio (class in neural_compressor.data.transforms.imagenet_transform)
ResizeWithRatio (class in neural_compressor.data.transforms.transform)
reverted_data_type() (in module neural_compressor.strategy.utils.utility)
RMSE (class in neural_compressor.metric.metric)
ROC (class in neural_compressor.metric.metric)
RoundingMode (class in neural_compressor.torch.algorithms.mx_quant.utils)
rtn_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
RTNConfig (class in neural_compressor.torch.quantization.config)
RTNQuantizer (class in neural_compressor.torch.algorithms.weight_only.rtn)
S
sample (neural_compressor.metric.metric.Accuracy attribute)
(neural_compressor.metric.metric.Loss attribute)
Sampler (class in neural_compressor.common.base_tuning)
save() (in module neural_compressor.torch.algorithms.pt2e_quant.save_load)
(in module neural_compressor.torch.algorithms.static_quant.save_load)
(in module neural_compressor.torch.algorithms.weight_only.save_load)
(in module neural_compressor.torch.quantization.save_load_entry)
save_config_mapping() (in module neural_compressor.common.utils.save_load)
save_for_huggingface_upstream() (in module neural_compressor.utils.load_huggingface)
save_layers_in_shards_iteratively() (in module neural_compressor.torch.algorithms.layer_wise.utils)
saved_model_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
SaveLoadFormat (class in neural_compressor.torch.utils.constants)
ScaleProPagationTransformer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation)
search_clip() (in module neural_compressor.torch.algorithms.weight_only.utility)
search_pattern (neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter.PatternPair attribute)
secure_check_eval_func() (in module neural_compressor.utils.utility)
SelfKnowledgeDistillationLossConfig (class in neural_compressor.config)
SequentialSampler (class in neural_compressor.common.base_tuning)
(class in neural_compressor.tensorflow.utils.data)
set_module() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
(in module neural_compressor.torch.algorithms.weight_only.utility)
(in module neural_compressor.torch.utils.utility)
set_random_seed() (in module neural_compressor.common.utils.utility)
(in module neural_compressor.utils.utility)
set_resume_from() (in module neural_compressor.common.utils.utility)
(in module neural_compressor.utils.utility)
set_tensorboard() (in module neural_compressor.common.utils.utility)
(in module neural_compressor.utils.utility)
set_workspace() (in module neural_compressor.common.utils.utility)
(in module neural_compressor.utils.utility)
shape (neural_compressor.torch.algorithms.weight_only.hqq.qtensor.QTensorMetaInfo attribute)
ShareQDQForItexYPatternOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern)
show_memory_info() (in module neural_compressor.utils.utility)
simple_inference() (in module neural_compressor.torch.algorithms.static_quant.utility)
singleton() (in module neural_compressor.common.utils.utility)
(in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
slim_session() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
smooth_quant_entry() (in module neural_compressor.tensorflow.quantization.algorithm_entry)
(in module neural_compressor.torch.quantization.algorithm_entry)
SmoothQuant (class in neural_compressor.algorithm.smooth_quant)
(class in neural_compressor.tensorflow.algorithms.smoother.core)
SmoothQuantCalibration (class in neural_compressor.tensorflow.algorithms.smoother.calibration)
SmoothQuantCalibrationLLM (class in neural_compressor.tensorflow.algorithms.smoother.calibration)
SmoothQuantConfig (class in neural_compressor.tensorflow.quantization.config)
(class in neural_compressor.torch.quantization.config)
SmoothQuantQuantizer (class in neural_compressor.torch.algorithms.smooth_quant.smooth_quant)
SmoothQuantSampler (class in neural_compressor.strategy.utils.tuning_sampler)
SmoothQuantScaler (class in neural_compressor.tensorflow.algorithms.smoother.scaler)
SmoothQuantScalerLLM (class in neural_compressor.tensorflow.algorithms.smoother.scaler)
SparseDummyDataset (class in neural_compressor.data.datasets.dummy_dataset_v2)
SplitSharedInputOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input)
SQLinearWrapper (class in neural_compressor.torch.algorithms.smooth_quant.utility)
SquadExample (class in neural_compressor.data.transforms.transform)
SquadF1 (class in neural_compressor.metric.metric)
static_quant_entry() (in module neural_compressor.tensorflow.quantization.algorithm_entry)
(in module neural_compressor.torch.quantization.algorithm_entry)
static_quant_export() (in module neural_compressor.utils.export.torch2onnx)
StaticQuantConfig (class in neural_compressor.tensorflow.keras.quantization.config)
(class in neural_compressor.tensorflow.quantization.config)
(class in neural_compressor.torch.quantization.config)
StaticQuantQuantizer (class in neural_compressor.torch.algorithms.static_quant.static_quant)
Statistics (class in neural_compressor.common.utils.utility)
(class in neural_compressor.utils.utility)
str2array() (in module neural_compressor.utils.utility)
strategy_registry() (in module neural_compressor.strategy.strategy)
strip_equivalent_nodes() (in module neural_compressor.tensorflow.quantization.utils.utility)
strip_unused_nodes() (in module neural_compressor.tensorflow.quantization.utils.utility)
StripEquivalentNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes)
StripUnusedNodesOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes)
StyleTransferDataset (class in neural_compressor.data.datasets.style_transfer_dataset)
sum (neural_compressor.metric.metric.Loss attribute)
SwitchOptimizer (class in neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer)
symbol_re (neural_compressor.metric.bleu.UnicodeRegex attribute)
T
TargetSpace (class in neural_compressor.strategy.bayesian)
TensorFlow (class in neural_compressor.config)
Tensorflow (class in neural_compressor.data.datasets.dataset)
Tensorflow_ITEXAdaptor (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow)
TensorFlowAdaptor (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow)
TensorflowBaseModel (class in neural_compressor.model.tensorflow_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
TensorflowBertDataset (class in neural_compressor.data.datasets.bert_dataset)
TensorflowCheckpointModel (class in neural_compressor.model.tensorflow_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
TensorflowCIFAR10 (class in neural_compressor.data.datasets.dataset)
TensorflowCIFAR100 (class in neural_compressor.data.datasets.dataset)
TensorflowCOCOMAP (class in neural_compressor.metric.metric)
TensorFlowConfig (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow)
TensorflowConfigConverter (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow)
TensorflowCropToBoundingBox (class in neural_compressor.data.transforms.transform)
TensorflowDatasets (class in neural_compressor.data.datasets.dataset)
TensorflowFashionMNIST (class in neural_compressor.data.datasets.dataset)
TensorflowFilters (class in neural_compressor.data.filters.filter)
TensorflowGlobalConfig (class in neural_compressor.tensorflow.utils.model)
TensorflowImagenetDataset (class in neural_compressor.data.datasets.imagenet_dataset)
TensorflowImagenetRaw (class in neural_compressor.data.datasets.imagenet_dataset)
TensorflowImageRecord (class in neural_compressor.data.datasets.dataset)
TensorflowLLMModel (class in neural_compressor.model.tensorflow_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
TensorflowMAP (class in neural_compressor.metric.metric)
TensorflowMetrics (class in neural_compressor.metric.metric)
TensorflowMNIST (class in neural_compressor.data.datasets.dataset)
TensorflowModel (class in neural_compressor.model.tensorflow_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
TensorflowModelZooBertDataset (class in neural_compressor.data.datasets.bert_dataset)
TensorflowQATModel (class in neural_compressor.model.tensorflow_model)
TensorflowQuery (class in neural_compressor.tensorflow.algorithms.static_quant.tensorflow)
TensorflowRandomHorizontalFlip (class in neural_compressor.data.transforms.transform)
TensorflowRandomVerticalFlip (class in neural_compressor.data.transforms.transform)
TensorflowResizeCropImagenetTransform (class in neural_compressor.data.transforms.imagenet_transform)
TensorflowResizeWithRatio (class in neural_compressor.data.transforms.transform)
TensorflowSavedModelModel (class in neural_compressor.model.tensorflow_model)
(class in neural_compressor.tensorflow.utils.model_wrappers)
TensorflowShiftRescale (class in neural_compressor.data.transforms.imagenet_transform)
TensorflowTFRecordDataset (class in neural_compressor.data.datasets.dataset)
TensorflowTopK (class in neural_compressor.metric.metric)
TensorflowTransform (class in neural_compressor.data.transforms.transform)
TensorflowTransforms (class in neural_compressor.data.transforms.transform)
TensorflowTranspose (class in neural_compressor.data.transforms.transform)
TensorflowTransposeLastChannel (class in neural_compressor.data.transforms.imagenet_transform)
TensorflowVOCMAP (class in neural_compressor.metric.metric)
TensorflowVOCRecord (class in neural_compressor.data.datasets.dataset)
TensorflowWrapFunction (class in neural_compressor.data.transforms.transform)
TensorQuantizer (class in neural_compressor.torch.algorithms.qat.tensor_quantizer)
teq_quantize_entry() (in module neural_compressor.torch.quantization.algorithm_entry)
TEQConfig (class in neural_compressor.torch.quantization.config)
TEQLinearFakeQuant (class in neural_compressor.torch.algorithms.weight_only.modules)
TEQuantizer (class in neural_compressor.torch.algorithms.weight_only.teq)
TF2ONNXConfig (class in neural_compressor.config)
tf_to_fp32_onnx() (in module neural_compressor.utils.export.tf2onnx)
tf_to_int8_onnx() (in module neural_compressor.utils.export.tf2onnx)
TFModelZooCollectTransform (class in neural_compressor.data.transforms.transform)
TFSlimNetsFactory (class in neural_compressor.model.nets_factory)
(class in neural_compressor.tensorflow.utils.utility)
TFSquadV1ModelZooPostTransform (class in neural_compressor.data.transforms.transform)
TFSquadV1PostTransform (class in neural_compressor.data.transforms.transform)
time_limit() (in module neural_compressor.utils.utility)
to_device() (in module neural_compressor.torch.utils.utility)
to_dtype() (in module neural_compressor.torch.utils.utility)
ToArray (class in neural_compressor.data.transforms.transform)
Torch2ONNXConfig (class in neural_compressor.config)
torch_to_fp32_onnx() (in module neural_compressor.utils.export.torch2onnx)
torch_to_int8_onnx() (in module neural_compressor.utils.export.torch2onnx)
TorchBaseConfig (class in neural_compressor.torch.quantization.config)
TorchSmoothQuant (class in neural_compressor.torch.algorithms.smooth_quant.utility)
TpeTuneStrategy (class in neural_compressor.contrib.strategy.tpe)
trace_gptq_target_blocks() (in module neural_compressor.torch.algorithms.weight_only.gptq)
TrainableEquivalentTransformation (class in neural_compressor.torch.algorithms.weight_only.teq)
transform_registry() (in module neural_compressor.data.transforms.transform)
transformation() (in module neural_compressor.torch.algorithms.pt2e_quant.half_precision_rewriter)
TransformerBasedModelBlockPatternDetector (class in neural_compressor.torch.algorithms.static_quant.utility)
TRANSFORMS (class in neural_compressor.data.transforms.transform)
Transforms (class in neural_compressor.data.transforms.transform)
Transpose (class in neural_compressor.data.transforms.transform)
try_loading_keras() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
TuneStrategy (class in neural_compressor.strategy.strategy)
TuneStrategyMeta (class in neural_compressor.strategy.strategy)
TuningConfig (class in neural_compressor.common.base_tuning)
TuningCriterion (class in neural_compressor.config)
TuningItem (class in neural_compressor.strategy.utils.tuning_space)
TuningLogger (class in neural_compressor.common.utils.logger)
TuningMonitor (class in neural_compressor.common.base_tuning)
TuningOrder (class in neural_compressor.strategy.utils.tuning_sampler)
TuningParam (class in neural_compressor.common.tuning_param)
TuningSampler (class in neural_compressor.strategy.utils.tuning_sampler)
TuningSpace (class in neural_compressor.strategy.utils.tuning_space)
U
UnicodeRegex (class in neural_compressor.metric.bleu)
UnpackedWeightOnlyLinearParams (class in neural_compressor.torch.algorithms.weight_only.modules)
UnpicklingError
update_module() (in module neural_compressor.torch.algorithms.layer_wise.utils)
update_sq_scale() (in module neural_compressor.torch.algorithms.smooth_quant.utility)
V
valid_keras_format() (in module neural_compressor.tensorflow.utils.utility)
valid_reshape_inputs() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
validate_and_inference_input_output() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
validate_graph_node() (in module neural_compressor.model.tensorflow_model)
(in module neural_compressor.tensorflow.utils.model_wrappers)
validate_modules() (in module neural_compressor.torch.utils.utility)
values_from_const() (in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in)
(in module neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm)
version1_eq_version2() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
version1_gt_version2() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
version1_gte_version2() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
version1_lt_version2() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
version1_lte_version2() (in module neural_compressor.tensorflow.utils.utility)
(in module neural_compressor.utils.utility)
W
W8A8PT2EQuantizer (class in neural_compressor.torch.algorithms.pt2e_quant.core)
warn() (in module neural_compressor.utils.logger)
warning() (in module neural_compressor.utils.logger)
WeightCorrection (class in neural_compressor.algorithm.weight_correction)
WeightOnlyLinear (class in neural_compressor.torch.algorithms.weight_only.modules)
WeightOnlyQuantSampler (class in neural_compressor.strategy.utils.tuning_sampler)
WeightPruningConfig (class in neural_compressor.config)
WeightsDetails (class in neural_compressor.utils.weights_details)
WeightsStatistics (class in neural_compressor.utils.weights_details)
whitespace_tokenize() (in module neural_compressor.data.transforms.tokenization)
WOQModelLoader (class in neural_compressor.torch.algorithms.weight_only.save_load)
WordpieceTokenizer (class in neural_compressor.data.transforms.tokenization)
WrapONNXRTMetric (class in neural_compressor.metric.metric)
WrapperLayer (class in neural_compressor.torch.algorithms.smooth_quant.utility)
WrapPyTorchMetric (class in neural_compressor.metric.metric)
write_graph() (in module neural_compressor.tensorflow.quantization.utils.utility)
write_json_file() (in module neural_compressor.torch.utils.utility)
X
XPU_Accelerator (class in neural_compressor.torch.utils.auto_accelerator)