Index

A

ActivationType (class in aimet_common.defs) analytical_bias_correction_per_layer() (in module aimet_tensorflow.bias_correction.BiasCorrection)

auto (aimet_tensorflow.defs.ChannelPruningParameters.Mode attribute) (aimet_tensorflow.defs.SpatialSvdParameters.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] BiasCorrectionParams() (in module aimet_tensorflow.bias_correction)

C

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

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

E

equalize_model() (in module aimet_tensorflow.cross_layer_equalization)

export() (aimet_tensorflow.quantsim.QuantizationSimModel method)

F

fold_all_batch_norms() (in module aimet_tensorflow.batch_norm_fold)

fold_given_batch_norms() (in module aimet_tensorflow.batch_norm_fold)

G

GreedySelectionParameters (class in aimet_common.defs)

L

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)

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) ModuleCompRatioPair (class in aimet_tensorflow.defs)

N

no_activation (aimet_common.defs.ActivationType attribute)

P

post_training_tf (aimet_common.defs.QuantScheme attribute)

post_training_tf_enhanced (aimet_common.defs.QuantScheme attribute)

Q

QuantizationSimModel (class in aimet_tensorflow.quantsim)

QuantParams (class in aimet_tensorflow.bias_correction) QuantScheme (class in aimet_common.defs)

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_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) scale_model() (in module aimet_tensorflow.cross_layer_equalization.CrossLayerScaling) spatial_svd (aimet_common.defs.CompressionScheme attribute)

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

V

visualize_relative_weight_ranges_single_layer() (in module aimet_tensorflow.plotting_utils)

visualize_weight_ranges_single_layer() (in module aimet_tensorflow.plotting_utils)

W

weight_svd (aimet_common.defs.CompressionScheme attribute)