Stay organized with collections Save and categorize content based on your preferences.

Specifies to prune only 1x1 Conv2D layers in the model.

Inherits From: PruningPolicy

Used in the notebooks

Used in the guide

PruneForLatencyOnXNNPack checks that the model contains a subgraph that can leverage XNNPACK's sparse inference and applies pruning wrapper only to Conv2D with kernel_size = (1, 1).



View source

Allows to prune only 1x1 Conv2D layers.


View source

Ensures that the model contains only supported layers.