neural_compressor.compression.pruner.patterns

Pruners.

Submodules

• neural_compressor.compression.pruner.patterns.base
• neural_compressor.compression.pruner.patterns.mha
• neural_compressor.compression.pruner.patterns.ninm
• neural_compressor.compression.pruner.patterns.nxm

Functions

get_pattern(config, modules[, framework])

Get registered pattern class.

Package Contents

neural_compressor.compression.pruner.patterns.get_pattern(config, modules, framework='pytorch')

Get registered pattern class.

Get a Pattern object from PATTERNS.

Parameters:

• config – A config dict object that contains the pattern information.

• modules – Torch neural network modules to be pruned with the pattern.

Returns:

A Pattern object.

Raises:

AssertionError – Currently only support patterns which have been registered in PATTERNS.