CompressionReference.md 861 Bytes
Newer Older
Ningxin Zheng's avatar
Ningxin Zheng committed
1
2
3
4
5
6
7
8
9
# Python API Reference of Compression Utilities

```eval_rst
.. contents::
```

## Sensitivity Utilities

```eval_rst
10
..  autoclass:: nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis
Ningxin Zheng's avatar
Ningxin Zheng committed
11
12
13
14
15
16
17
    :members:

```

## Topology Utilities

```eval_rst
18
..  autoclass:: nni.compression.pytorch.utils.shape_dependency.ChannelDependency
Ningxin Zheng's avatar
Ningxin Zheng committed
19
20
    :members:

21
..  autoclass:: nni.compression.pytorch.utils.shape_dependency.GroupDependency
Ningxin Zheng's avatar
Ningxin Zheng committed
22
    :members:
23

24
..  autoclass:: nni.compression.pytorch.utils.mask_conflict.CatMaskPadding
25
26
    :members:

27
..  autoclass:: nni.compression.pytorch.utils.mask_conflict.GroupMaskConflict
28
29
    :members:

30
..  autoclass:: nni.compression.pytorch.utils.mask_conflict.ChannelMaskConflict
31
32
    :members:

Ningxin Zheng's avatar
Ningxin Zheng committed
33
```
34
35
36
37

## Model FLOPs/Parameters Counter

```eval_rst
38
..  autofunction:: nni.compression.pytorch.utils.counter.count_flops_params
39

40
```