Index

_

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

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

A

add() (Mapping method)

apply() (ChannelsSelectorConstraint method)

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)

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)

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

LossAwareCriterion (class in enot.pruning_v2)

M

Mapping (class in enot.distillation) max_iterations (KnapsackLabelSelector property)

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

N

n_prunable_groups (ModelPruningInfo property)

O

OpenVINOFakeQuantizedModel (class in enot.quantization)

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

P

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

ScorePruningLabelSelector (class in enot.pruning) select() (GlobalScoreLabelSelector method) (KnapsackLabelSelector method) (PruningLabelSelector method) (ScorePruningLabelSelector method) (UniformLabelSelector 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)