File tree
689 files changed
+760
-729
lines changed- latest
- _sources/autoapi/neural_compressor
- adaptor
- mxnet
- onnxrt
- tf_utils/quantize_graph/quantize_graph_concatv2
- compression/pruner/pruners/basic
- data/dataloaders
- mxnet_dataloader
- onnxrt_dataloader
- tensorflow_dataloader
- experimental
- data/dataloaders
- mxnet_dataloader
- onnxrt_dataloader
- tensorflow_dataloader
- strategy
- basic
- exhaustive
- random
- strategy
- basic
- exhaustive
- random
- torch/quantization/config
- autoapi
- block_mask
- neural_compressor
- adaptor
- adaptor
- keras
- keras_utils
- conv2d
- dense
- depthwise_conv2d
- pool2d
- quantizer
- separable_conv2d
- mxnet
- mxnet_utils
- util
- onnxrt
- ox_utils
- calibration
- calibrator
- operators
- activation
- argmax
- attention
- binary_op
- concat
- conv
- direct_q8
- embed_layernorm
- gather
- gavgpool
- gemm
- lstm
- matmul
- maxpool
- norm
- ops
- pad
- pooling
- reduce
- resize
- split
- unary_op
- quantizer
- smooth_quant
- util
- weight_only
- pytorch
- query
- tensorflow
- tf_utils
- graph_converter
- graph_converter_without_calib
- graph_rewriter
- bf16
- bf16_convert
- dequantize_cast_optimizer
- generic
- convert_add_to_biasadd
- convert_layout
- convert_leakyrelu
- convert_nan_to_random
- convert_placeholder_to_const
- dilated_contraction
- dummy_biasadd
- expanddims_optimizer
- fetch_weight_from_reshape
- fold_batch_norm
- fold_constant
- fuse_biasadd_add
- fuse_column_wise_mul
- fuse_conv_with_math
- fuse_decomposed_bn
- fuse_decomposed_in
- fuse_gelu
- fuse_layer_norm
- fuse_pad_with_conv
- fuse_pad_with_fp32_conv
- fuse_reshape_transpose
- graph_cse_optimizer
- grappler_pass
- insert_print_node
- move_squeeze_after_relu
- pre_optimize
- remove_training_nodes
- rename_batch_norm
- strip_equivalent_nodes
- strip_unused_nodes
- switch_optimizer
- graph_base
- int8
- freeze_fake_quant
- freeze_value
- freeze_value_without_calib
- fuse_conv_redundant_dequantize
- fuse_conv_requantize
- fuse_matmul_redundant_dequantize
- fuse_matmul_requantize
- meta_op_optimizer
- post_hostconst_converter
- post_quantized_op_cse
- rnn_convert
- scale_propagation
- onnx
- onnx_graph
- onnx_node
- onnx_schema
- tf2onnx_utils
- qdq
- insert_qdq_pattern
- merge_duplicated_qdq
- graph_util
- quantize_graph
- qat
- fake_quantize
- quantize_config
- quantize_helper
- quantize_layers
- optimize_layer
- quantize_layer_add
- quantize_layer_base
- quantize_layer_bn
- quantize_wrapper
- qdq
- fuse_qdq_bn
- fuse_qdq_concatv2
- fuse_qdq_conv
- fuse_qdq_deconv
- fuse_qdq_in
- fuse_qdq_matmul
- fuse_qdq_pooling
- optimize_qdq
- quantize_graph_base
- quantize_graph_bn
- quantize_graph_concatv2
- quantize_graph_conv
- quantize_graph_for_intel_cpu
- quantize_graph_matmul
- quantize_graph_pooling
- quantize_graph_common
- smooth_quant_calibration
- smooth_quant_scaler
- tf2onnx_converter
- transform_graph
- bias_correction
- graph_transform_base
- insert_logging
- rerange_quantized_concat
- util
- torch_utils
- autoround
- autoround
- export
- model_wrapper
- sign_sgd
- awq
- bf16_convert
- gptq
- hawq_metric
- layer_wise_quant
- modified_pickle
- quantize
- torch_load
- utils
- mixed_precision
- model_wrapper
- pattern_detector
- smooth_quant
- symbolic_trace
- teq
- util
- weight_only
- algorithm
- algorithm
- fast_bias_correction
- smooth_quant
- weight_correction
- benchmark
- common
- base_config
- base_tuning
- tuning_param
- utils
- constants
- logger
- utility
- compression
- callbacks
- distillation
- criterions
- optimizers
- utility
- hpo
- sa_optimizer
- search_algorithms
- search_space
- pruner
- criteria
- dsnot
- model_slim
- auto_slim
- pattern_analyzer
- weight_slim
- patterns
- base
- mha
- ninm
- nxm
- pruners
- base
- basic
- block_mask
- mha
- pattern_lock
- progressive
- retrain_free
- sparse_gpt
- pruning
- regs
- schedulers
- tf_criteria
- utils
- wanda
- prune
- utils
- wrapper
- conf
- config
- dotdict
- pythonic_config
- config
- contrib
- strategy
- sigopt
- tpe
- data
- dataloaders
- base_dataloader
- dataloader
- default_dataloader
- fetcher
- mxnet_dataloader
- onnxrt_dataloader
- pytorch_dataloader
- sampler
- tensorflow_dataloader
- datasets
- bert_dataset
- coco_dataset
- dataset
- dummy_dataset
- dummy_dataset_v2
- imagenet_dataset
- style_transfer_dataset
- filters
- coco_filter
- filter
- transforms
- coco_transform
- imagenet_transform
- postprocess
- tokenization
- transform
- experimental
- benchmark
- common
- criterion
- dataloader
- metric
- model
- optimizer
- postprocess
- torch_utils
- component
- compression
- contrib
- strategy
- sigopt
- tpe
- data
- dataloaders
- base_dataloader
- dataloader
- default_dataloader
- fetcher
- mxnet_dataloader
- onnxrt_dataloader
- pytorch_dataloader
- sampler
- tensorflow_dataloader
- datasets
- bert_dataset
- coco_dataset
- dataset
- dummy_dataset
- dummy_dataset_v2
- imagenet_dataset
- style_transfer_dataset
- filters
- coco_filter
- filter
- transforms
- imagenet_transform
- tokenization
- transform
- distillation
- export
- qlinear2qdq
- tf2onnx
- torch2onnx
- graph_optimization
- metric
- bleu
- bleu_util
- coco_label_map
- coco_tools
- evaluate_squad
- f1
- metric
- mixed_precision
- model_conversion
- nas
- basic_nas
- dynas
- nas
- nas_utils
- search_algorithms
- pruner_legacy
- gradient_sensitivity
- group_lasso
- magnitude
- pattern_lock
- pruner
- pruning
- pruning_recipes
- patterns
- pattern
- tile_pattern
- pruning_v2
- pytorch_pruner
- logger
- patterns
- prune_utils
- pruner
- pruning
- scheduler
- quantization
- scheduler
- strategy
- auto_mixed_precision
- basic
- bayesian
- exhaustive
- mse
- mse_v2
- random
- strategy
- utils
- constant
- tuning_sampler
- tuning_space
- tuning_structs
- utility
- metric
- bleu
- bleu_util
- coco_label_map
- coco_tools
- evaluate_squad
- f1
- metric
- mix_precision
- model
- base_model
- keras_model
- model
- mxnet_model
- nets_factory
- onnx_model
- tensorflow_model
- torch_model
- objective
- onnxrt
- algorithms
- smoother
- calibrator
- core
- weight_only
- awq
- gptq
- rtn
- utility
- quantization
- algorithm_entry
- autotune
- calibrate
- config
- quantize
- utils
- onnx_model
- utility
- profiling
- parser
- factory
- onnx_parser
- factory
- parser
- parser
- result
- tensorflow_parser
- factory
- parser
- profiler
- factory
- onnxrt_profiler
- factory
- profiler
- utils
- profiler
- tensorflow_profiler
- factory
- profiler
- utils
- quantization
- strategy
- auto
- auto_mixed_precision
- basic
- bayesian
- conservative
- exhaustive
- hawq_v2
- mse
- mse_v2
- random
- strategy
- utils
- constant
- tuning_sampler
- tuning_space
- tuning_structs
- utility
- template
- api_doc_example
- tensorflow
- algorithms
- smoother
- calibration
- core
- scaler
- static_quant
- keras
- keras_utils
- conv2d
- dense
- depthwise_conv2d
- pool2d
- quantizer
- separable_conv2d
- quantization
- algorithm_entry
- config
- quantize
- utils
- graph_rewriter
- generic
- convert_add_to_biasadd
- convert_layout
- convert_leakyrelu
- convert_nan_to_random
- convert_placeholder_to_const
- dilated_contraction
- dummy_biasadd
- expanddims_optimizer
- fetch_weight_from_reshape
- fold_batch_norm
- fold_constant
- fuse_biasadd_add
- fuse_column_wise_mul
- fuse_conv_with_math
- fuse_decomposed_bn
- fuse_decomposed_in
- fuse_gelu
- fuse_layer_norm
- fuse_pad_with_conv
- fuse_pad_with_fp32_conv
- fuse_reshape_transpose
- graph_cse_optimizer
- grappler_pass
- insert_print_node
- move_squeeze_after_relu
- pre_optimize
- remove_training_nodes
- rename_batch_norm
- strip_equivalent_nodes
- strip_unused_nodes
- switch_optimizer
- graph_base
- graph_util
- quantize_graph_common
- utility
- utils
- constants
- data
- model
- model_wrappers
- nets_factory
- utility
- torch
- algorithms
- habana_fp8
- fp8_quant
- modules
- observer
- layer_wise
- load
- modified_pickle
- utils
- static_quant
- static_quant
- utility
- weight_only
- awq
- gptq
- hqq
- auto_accelerator
- bitpack
- config
- core
- optimizer
- qtensor
- quant_api
- quantizer
- utility
- modules
- rtn
- utility
- amp
- autocast
- fp8
- functions
- quantization
- algorithm_entry
- autotune
- config
- modules
- quantize
- utils
- constants
- environ
- utility
- training
- utils
- collect_layer_histogram
- constant
- create_obj_from_config
- kl_divergence
- load_huggingface
- logger
- neural_insights_utils
- options
- pytorch
- utility
- weights_details
- version
- docs
- build_docs/source
- source
- api-doc
- adaptor
- neural_coder
- docs
- release_notes
- extensions
- neural_compressor_ext_lab
- neural_compressor_ext_lab_alibaba
- neural_compressor_ext_vscode
- neural_insights
- docs/source
- gui
- neural_solution
- docs/source
- template
- examples
- custom_models_optimized/tf_example1
- hf_models
- hf_models_grpc
- frontend
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
689 files changed
+760
-729
lines changedOriginal file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
15 | 15 |
| |
16 | 16 |
| |
17 | 17 |
| |
| 18 | + | |
18 | 19 |
| |
19 | 20 |
| |
20 | 21 |
| |
| |||
30 | 31 |
| |
31 | 32 |
| |
32 | 33 |
| |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + |
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
19 | 19 |
| |
20 | 20 |
| |
21 | 21 |
| |
| 22 | + | |
22 | 23 |
| |
23 | 24 |
| |
24 | 25 |
| |
| |||
78 | 79 |
| |
79 | 80 |
| |
80 | 81 |
| |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
| 90 | + | |
| 91 | + |
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
54 | 54 |
| |
55 | 55 |
| |
56 | 56 |
| |
57 |
| - | |
| 57 | + | |
58 | 58 |
| |
59 | 59 |
| |
60 | 60 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
23 | 23 |
| |
24 | 24 |
| |
25 | 25 |
| |
26 |
| - | |
| 26 | + | |
27 | 27 |
| |
28 | 28 |
| |
29 | 29 |
| |
30 | 30 |
| |
31 | 31 |
| |
32 | 32 |
| |
33 | 33 |
| |
34 |
| - | |
| 34 | + | |
35 | 35 |
| |
36 | 36 |
| |
37 | 37 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
40 | 40 |
| |
41 | 41 |
| |
42 | 42 |
| |
43 |
| - | |
| 43 | + | |
44 | 44 |
| |
45 | 45 |
| |
46 | 46 |
| |
| |||
50 | 50 |
| |
51 | 51 |
| |
52 | 52 |
| |
53 |
| - | |
| 53 | + | |
54 | 54 |
| |
55 | 55 |
| |
56 | 56 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
23 | 23 |
| |
24 | 24 |
| |
25 | 25 |
| |
26 |
| - | |
| 26 | + | |
27 | 27 |
| |
28 | 28 |
| |
29 | 29 |
| |
30 | 30 |
| |
31 | 31 |
| |
32 | 32 |
| |
33 | 33 |
| |
34 |
| - | |
| 34 | + | |
35 | 35 |
| |
36 | 36 |
| |
37 | 37 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
40 | 40 |
| |
41 | 41 |
| |
42 | 42 |
| |
43 |
| - | |
| 43 | + | |
44 | 44 |
| |
45 | 45 |
| |
46 | 46 |
| |
| |||
50 | 50 |
| |
51 | 51 |
| |
52 | 52 |
| |
53 |
| - | |
| 53 | + | |
54 | 54 |
| |
55 | 55 |
| |
56 | 56 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
22 | 22 |
| |
23 | 23 |
| |
24 | 24 |
| |
25 |
| - | |
| 25 | + | |
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
64 | 64 |
| |
65 | 65 |
| |
66 | 66 |
| |
67 |
| - | |
68 |
| - | |
69 |
| - | |
70 |
| - | |
71 |
| - | |
72 |
| - | |
73 |
| - | |
74 | 67 |
| |
75 | 68 |
| |
76 | 69 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
107 | 107 |
| |
108 | 108 |
| |
109 | 109 |
| |
110 |
| - | |
| 110 | + | |
111 | 111 |
| |
112 | 112 |
| |
113 | 113 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
146 | 146 |
| |
147 | 147 |
| |
148 | 148 |
| |
149 |
| - | |
| 149 | + | |
150 | 150 |
| |
151 | 151 |
| |
152 | 152 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
217 | 217 |
| |
218 | 218 |
| |
219 | 219 |
| |
220 |
| - | |
| 220 | + | |
221 | 221 |
| |
222 | 222 |
| |
223 | 223 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
125 | 125 |
| |
126 | 126 |
| |
127 | 127 |
| |
128 |
| - | |
| 128 | + | |
129 | 129 |
| |
130 | 130 |
| |
131 | 131 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
106 | 106 |
| |
107 | 107 |
| |
108 | 108 |
| |
109 |
| - | |
| 109 | + | |
110 | 110 |
| |
111 | 111 |
| |
112 | 112 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
99 | 99 |
| |
100 | 100 |
| |
101 | 101 |
| |
| 102 | + | |
| 103 | + | |
| 104 | + | |
102 | 105 |
| |
103 | 106 |
| |
104 | 107 |
| |
| |||
112 | 115 |
| |
113 | 116 |
| |
114 | 117 |
| |
| 118 | + | |
| 119 | + | |
| 120 | + | |
| 121 | + | |
| 122 | + | |
| 123 | + | |
| 124 | + | |
115 | 125 |
| |
116 | 126 |
| |
117 | 127 |
| |
| |||
130 | 140 |
| |
131 | 141 |
| |
132 | 142 |
| |
133 |
| - | |
| 143 | + | |
134 | 144 |
| |
135 | 145 |
| |
136 | 146 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
115 | 115 |
| |
116 | 116 |
| |
117 | 117 |
| |
118 |
| - | |
| 118 | + | |
119 | 119 |
| |
120 | 120 |
| |
121 | 121 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
679 | 679 |
| |
680 | 680 |
| |
681 | 681 |
| |
682 |
| - | |
| 682 | + | |
683 | 683 |
| |
684 | 684 |
| |
685 | 685 |
| |
|
0 commit comments