Index

_

__init__() (calibrate method) (DefaultChannelsSelectorConstraint method) (distill method), [1] (GlobalLatencyPruningLabelSelector method) (GlobalPruningLabelSelectorByChannels method) (KnapsackPruningLabelSelector method) (Mapping method) (OpenVINOFakeQuantizedModel method) (OptimalPruningLabelSelector method) (PruningCalibrator method) (RescalingCalibrator method) (STMFakeQuantizedModel method) (TensorRTFakeQuantizedModel 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)

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)

I

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

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

K

KnapsackPruningLabelSelector (class in enot.pruning)

M

Mapping (class in enot.distillation) 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_model() (in module enot.pruning)

pruning_info (PruningCalibrator property) PruningCalibrator (class in enot.pruning) 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() (PruningLabelSelector method) (ScorePruningLabelSelector 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

UniformPruningLabelSelector (class in enot.pruning)