neural_compressor.experimental.pruning_recipes.patterns.tile_pattern

Tile pattern classes.

Module Contents

Classes

TilePatternBase

Parent class for all NxM tile patterns.

TilePattern_1x1

1x1 tile pattern (unstructured).

TilePattern_2x2

2x2 tile pattern (unstructured).

TilePattern_1x16

1x16 tile pattern (unstructured).

TilePattern_4x1

4x1 tile pattern (unstructured).

TilePattern_1x2

1x2 tile pattern (unstructured).

class neural_compressor.experimental.pruning_recipes.patterns.tile_pattern.TilePatternBase(mask_shape)[source]

Parent class for all NxM tile patterns.

class neural_compressor.experimental.pruning_recipes.patterns.tile_pattern.TilePattern_1x1[source]

1x1 tile pattern (unstructured).

class neural_compressor.experimental.pruning_recipes.patterns.tile_pattern.TilePattern_2x2[source]

2x2 tile pattern (unstructured).

class neural_compressor.experimental.pruning_recipes.patterns.tile_pattern.TilePattern_1x16[source]

1x16 tile pattern (unstructured).

class neural_compressor.experimental.pruning_recipes.patterns.tile_pattern.TilePattern_4x1[source]

4x1 tile pattern (unstructured).

class neural_compressor.experimental.pruning_recipes.patterns.tile_pattern.TilePattern_1x2[source]

1x2 tile pattern (unstructured).