Index

A

ActivationType (class in aimet_common.defs) AdaroundParameters (class in aimet_tensorflow.adaround.adaround_weight) (class in aimet_torch.adaround.adaround_weight) analytical_bias_correction_per_layer() (in module aimet_tensorflow.bias_correction.BiasCorrection) apply_adaround() (in module aimet_tensorflow.adaround.adaround_weight.Adaround) (in module aimet_torch.adaround.adaround_weight.Adaround)

auto (aimet_tensorflow.defs.ChannelPruningParameters.Mode attribute) (aimet_tensorflow.defs.SpatialSvdParameters.Mode attribute) (aimet_torch.defs.ChannelPruningParameters.Mode attribute) (aimet_torch.defs.SpatialSvdParameters.Mode attribute) (aimet_torch.defs.WeightSvdParameters.Mode attribute)

B

bias_correction_per_layer() (in module aimet_tensorflow.bias_correction.BiasCorrection) bias_fold() (in module aimet_tensorflow.cross_layer_equalization.HighBiasFold), [1] (in module aimet_tensorflow.keras.cross_layer_equalization.HighBiasFold), [1] (in module aimet_torch.cross_layer_equalization.HighBiasFold), [1]

BiasCorrectionParams() (in module aimet_tensorflow.bias_correction)

C

channel_pruning (aimet_common.defs.CompressionScheme attribute) ChannelPruningParameters (class in aimet_tensorflow.defs) (class in aimet_torch.defs) ChannelPruningParameters.AutoModeParams (class in aimet_tensorflow.defs) (class in aimet_torch.defs) ChannelPruningParameters.ManualModeParams (class in aimet_tensorflow.defs) (class in aimet_torch.defs) ChannelPruningParameters.Mode (class in aimet_tensorflow.defs) (class in aimet_torch.defs) ClsSetInfo (class in aimet_tensorflow.cross_layer_equalization) (class in aimet_tensorflow.keras.cross_layer_equalization) (class in aimet_torch.cross_layer_equalization) ClsSetInfo.ClsSetLayerPairInfo (class in aimet_tensorflow.cross_layer_equalization) (class in aimet_tensorflow.keras.cross_layer_equalization) (class in aimet_torch.cross_layer_equalization)

compress_model() (aimet_tensorflow.compress.ModelCompressor static method) (aimet_torch.compress.ModelCompressor static method) compress_net() (aimet_tensorflow.svd.Svd method) CompressionScheme (class in aimet_common.defs) compute_encodings() (aimet_tensorflow.keras.quantsim.QuantizationSimModel method) (aimet_tensorflow.quantsim.QuantizationSimModel method) (aimet_torch.quantsim.QuantizationSimModel method) ConvBnInfoType (class in aimet_common.bias_correction) correct_bias() (in module aimet_tensorflow.bias_correction.BiasCorrection) (in module aimet_torch.bias_correction) CostMetric (class in aimet_common.defs)

D

display_comp_ratio_plot() (aimet_torch.visualize_serialized_data.VisualizeCompression method)

display_eval_scores() (aimet_torch.visualize_serialized_data.VisualizeCompression method)

E

equalize_model() (in module aimet_tensorflow.cross_layer_equalization) (in module aimet_torch.cross_layer_equalization)

export() (aimet_tensorflow.keras.quantsim.QuantizationSimModel method) (aimet_tensorflow.quantsim.QuantizationSimModel method) (aimet_torch.quantsim.QuantizationSimModel method)

F

fold_all_batch_norms() (in module aimet_tensorflow.batch_norm_fold) (in module aimet_tensorflow.keras.batch_norm_fold) (in module aimet_torch.batch_norm_fold)

fold_given_batch_norms() (in module aimet_tensorflow.batch_norm_fold) (in module aimet_tensorflow.keras.batch_norm_fold) (in module aimet_torch.batch_norm_fold)

G

GreedySelectionParameters (class in aimet_common.defs)

L

load_checkpoint() (aimet_torch.quantsim method)

load_keras_model_multi_gpu() (in module aimet_tensorflow.utils.convert_tf_sess_to_keras) load_tf_sess_variables_to_keras_single_gpu() (in module aimet_tensorflow.utils.convert_tf_sess_to_keras)

M

mac (aimet_common.defs.CostMetric attribute) manual (aimet_tensorflow.defs.ChannelPruningParameters.Mode attribute) (aimet_tensorflow.defs.SpatialSvdParameters.Mode attribute) (aimet_torch.defs.ChannelPruningParameters.Mode attribute) (aimet_torch.defs.SpatialSvdParameters.Mode attribute) (aimet_torch.defs.WeightSvdParameters.Mode attribute)

map_cls_sets_to_new_session() (aimet_tensorflow.cross_layer_equalization.ClsSetInfo static method) memory (aimet_common.defs.CostMetric attribute) ModelCompressor (class in aimet_tensorflow.compress) (class in aimet_torch.compress) ModuleCompRatioPair (class in aimet_tensorflow.defs) (class in aimet_torch.defs)

N

no_activation (aimet_common.defs.ActivationType attribute)

P

post_training_tf (aimet_common.defs.QuantScheme attribute), [1], [2]

post_training_tf_enhanced (aimet_common.defs.QuantScheme attribute), [1], [2] prepare_model() (in module aimet_torch.model_preparer)

Q

QuantizationSimModel (class in aimet_tensorflow.keras.quantsim) (class in aimet_tensorflow.quantsim) (class in aimet_torch.quantsim)

QuantParams (class in aimet_tensorflow.bias_correction) (class in aimet_torch.quantsim) QuantScheme (class in aimet_common.defs), [1], [2]

R

relu (aimet_common.defs.ActivationType attribute)

relu6 (aimet_common.defs.ActivationType attribute)

S

save_as_tf_module_multi_gpu() (in module aimet_tensorflow.utils.convert_tf_sess_to_keras) save_checkpoint() (aimet_torch.quantsim method) save_tf_session_single_gpu() (in module aimet_tensorflow.utils.convert_tf_sess_to_keras) scale_cls_sets() (in module aimet_tensorflow.cross_layer_equalization.CrossLayerScaling) (in module aimet_tensorflow.keras.cross_layer_equalization.CrossLayerScaling) (in module aimet_torch.cross_layer_equalization.CrossLayerScaling) scale_model() (in module aimet_tensorflow.cross_layer_equalization.CrossLayerScaling) (in module aimet_torch.cross_layer_equalization.CrossLayerScaling) spatial_svd (aimet_common.defs.CompressionScheme attribute)

SpatialSvdParameters (class in aimet_tensorflow.defs) (class in aimet_torch.defs) SpatialSvdParameters.AutoModeParams (class in aimet_tensorflow.defs) (class in aimet_torch.defs) SpatialSvdParameters.ManualModeParams (class in aimet_tensorflow.defs) (class in aimet_torch.defs) SpatialSvdParameters.Mode (class in aimet_tensorflow.defs) (class in aimet_torch.defs) Svd (class in aimet_tensorflow.svd)

T

TarRankSelectionParameters (class in aimet_torch.defs)

training_range_learning_with_tf_enhanced_init (aimet_common.defs.QuantScheme attribute), [1], [2] training_range_learning_with_tf_init (aimet_common.defs.QuantScheme attribute), [1], [2]

V

visualize_changes_after_optimization() (in module aimet_torch.visualize_model) visualize_relative_weight_ranges_single_layer() (in module aimet_tensorflow.plotting_utils) visualize_relative_weight_ranges_to_identify_problematic_layers() (in module aimet_torch.visualize_model)

visualize_weight_ranges() (in module aimet_torch.visualize_model) visualize_weight_ranges_single_layer() (in module aimet_tensorflow.plotting_utils) VisualizeCompression (class in aimet_torch.visualize_serialized_data)

W

weight_svd (aimet_common.defs.CompressionScheme attribute) WeightSvdParameters (class in aimet_torch.defs)

WeightSvdParameters.AutoModeParams (class in aimet_torch.defs) WeightSvdParameters.ManualModeParams (class in aimet_torch.defs) WeightSvdParameters.Mode (class in aimet_torch.defs)