Index

_

__init__() (BinarySearchLatencyLabelSelector method) (calibrate method) (DefaultChannelsSelectorConstraint method) (distill method), [1] (GlobalLatencyPruningLabelSelector method) (GlobalPruningLabelSelectorByChannels method) (GlobalScoreLabelSelector method) (GTBaselineOptimizer method) (KnapsackLabelSelector method) (KnapsackPruningLabelSelector method) (LossAwareCriterion method) (Mapping method) (OpenVINOFakeQuantizedModel method) (OptimalPruningLabelSelector method) (PruningCalibrator method) (PruningIR method) (RescalingCalibrator method) (SCBOLabelSelector method) (STMFakeQuantizedModel method) (TensorRTFakeQuantizedModel method) (UniformLabelSelector method) (UniformPruningLabelSelector method)

_get() (ChannelsSelectorConstraint method) _select() (ScorePruningLabelSelector method)

A

add() (Mapping method) add_param_group() (GTBaselineOptimizer method)

apply() (ChannelsSelectorConstraint method) attention_pruning_cfg (PruningIR attribute)

B

batch_size (SCBOLabelSelector property)

BinarySearchLatencyLabelSelector (class in enot.pruning_v2)

C

calibrate (class in enot.quantization) calibrate_and_prune_model() (in module enot.pruning) calibrate_and_prune_model_equal() (in module enot.pruning)

calibrate_and_prune_model_optimal() (in module enot.pruning) calibrate_model_for_pruning() (in module enot.pruning) ChannelsSelectorConstraint (class in enot.pruning) conv2d_adapter() (in module enot.distillation.adapter)

D

default_sample_to_model_inputs() (in module enot.utils.dataloader2model) default_sample_to_n_samples() (in module enot.utils.dataloader2model) DefaultChannelsSelectorConstraint (class in enot.pruning)

distill (class in enot.distillation) (class in enot.quantization) DistillationLayerSelectionStrategy (class in enot.quantization)

E

enable_calibration_mode() (FakeQuantizedModel method)

enable_quantization_mode() (FakeQuantizedModel method) extract_labels_info() (PruningLabelSelector static method)

F

FakeQuantizedModel (class in enot.quantization)

float_model_from_quantized_model() (in module enot.quantization.utils) force_thunk_update (PruningIR attribute)

G

get() (ChannelsSelectorConstraint method) get_label_filters() (in module enot.pruning.tools) get_module_name_by_tracing_node() (in module enot.tensor_trace.tools) get_most_important_labels() (PruningLabelSelector static method)

get_tracing_node_by_module_name() (in module enot.tensor_trace.tools) GlobalLatencyPruningLabelSelector (class in enot.pruning) GlobalPruningLabelSelectorByChannels (class in enot.pruning) GlobalScoreLabelSelector (class in enot.pruning_v2) GTBaselineOptimizer (class in enot.optimization)

I

inverted_conv2d_adapter() (in module enot.distillation.adapter)

is_bn() (in module enot.utils.batch_norm)

K

KnapsackLabelSelector (class in enot.pruning_v2)

KnapsackPruningLabelSelector (class in enot.pruning)

L

LatencyMeasurementError (class in enot.pruning_v2.label_selector)

load_state_dict() (GTBaselineOptimizer method) LossAwareCriterion (class in enot.pruning_v2)

M

Mapping (class in enot.distillation) max_iterations (KnapsackLabelSelector property) max_relative_latency_step (KnapsackLabelSelector property) model (GTBaselineOptimizer property)

model_reset_bn() (in module enot.utils.batch_norm) model_set_bn_momentum() (in module enot.utils.batch_norm) model_step() (GTBaselineOptimizer method) ModelPruningInfo (class in enot.pruning)

N

n_prunable_groups (ModelPruningInfo property)

n_search_steps (SCBOLabelSelector property) n_starting_points (SCBOLabelSelector property)

O

OpenVINOFakeQuantizedModel (class in enot.quantization)

optimal_quantization_scheme() (in module enot.quantization.utils) OptimalPruningLabelSelector (class in enot.pruning)

P

param_groups (GTBaselineOptimizer property) payload() (Mapping method) prune() (PruningIR method) prune_model() (in module enot.pruning)

pruning_info (PruningCalibrator property) PruningCalibrator (class in enot.pruning) PruningIR (class in enot.pruning_v2) PruningLabelSelector (class in enot.pruning)

Q

QuantDistillationModule (class in enot.quantization)

quantization_parameters() (FakeQuantizedModel method)

R

regular_parameters() (FakeQuantizedModel method) rescale() (RescalingCalibrator method)

RescalingCalibrator (class in enot.rescaling) reset_bn() (in module enot.utils.batch_norm) RMSELoss (class in enot.quantization)

S

SCBOLabelSelector (class in enot.pruning_v2) ScorePruningLabelSelector (class in enot.pruning) select() (PruningLabelSelector method) (ScorePruningLabelSelector method)

set_search_step_cb() (SCBOLabelSelector method) state_dict() (GTBaselineOptimizer method) step() (GTBaselineOptimizer method) STMFakeQuantizedModel (class in enot.quantization) summary() (ModelPruningInfo method)

T

tabulate_label_dependencies() (in module enot.pruning.tools) tabulate_label_filters() (in module enot.pruning.tools) tabulate_module_dependencies() (in module enot.pruning.tools)

tabulate_unprunable_groups() (in module enot.pruning.tools) TensorRTFakeQuantizedModel (class in enot.quantization) tune_bn_stats() (in module enot.utils.batch_norm)

U

UniformLabelSelector (class in enot.pruning_v2)

UniformPruningLabelSelector (class in enot.pruning)

V

verbose (SCBOLabelSelector property)

Z

zero_grad() (GTBaselineOptimizer method)