neural_compressor.compression

Subpackages

• neural_compressor.compression.distillation
  • neural_compressor.compression.distillation.criterions
  • neural_compressor.compression.distillation.optimizers
  • neural_compressor.compression.distillation.utility
• neural_compressor.compression.hpo
  • neural_compressor.compression.hpo.sa_optimizer
• neural_compressor.compression.pruner
  • neural_compressor.compression.pruner.model_slim
    • neural_compressor.compression.pruner.model_slim.auto_slim
    • neural_compressor.compression.pruner.model_slim.pattern_analyzer
    • neural_compressor.compression.pruner.model_slim.weight_slim
  • neural_compressor.compression.pruner.patterns
    • neural_compressor.compression.pruner.patterns.base
    • neural_compressor.compression.pruner.patterns.mha
    • neural_compressor.compression.pruner.patterns.ninm
    • neural_compressor.compression.pruner.patterns.nxm
  • neural_compressor.compression.pruner.pruners
    • neural_compressor.compression.pruner.pruners.base
    • neural_compressor.compression.pruner.pruners.basic
    • neural_compressor.compression.pruner.pruners.block_mask
    • neural_compressor.compression.pruner.pruners.mha
    • neural_compressor.compression.pruner.pruners.pattern_lock
    • neural_compressor.compression.pruner.pruners.progressive
    • neural_compressor.compression.pruner.pruners.retrain_free
  • neural_compressor.compression.pruner.wanda
    • neural_compressor.compression.pruner.wanda.utils
  • neural_compressor.compression.pruner.criteria
  • neural_compressor.compression.pruner.pruning
  • neural_compressor.compression.pruner.regs
  • neural_compressor.compression.pruner.schedulers
  • neural_compressor.compression.pruner.tf_criteria
  • neural_compressor.compression.pruner.utils

Submodules

• neural_compressor.compression.callbacks