CompressionReference.md 846 Bytes
Newer Older
Ningxin Zheng's avatar
Ningxin Zheng committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Python API Reference of Compression Utilities

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

## Sensitivity Utilities

```eval_rst
..  autoclass:: nni.compression.torch.utils.sensitivity_analysis.SensitivityAnalysis
    :members:

```

## Topology Utilities

```eval_rst
..  autoclass:: nni.compression.torch.utils.shape_dependency.ChannelDependency
    :members:

21
..  autoclass:: nni.compression.torch.utils.shape_dependency.GroupDependency
Ningxin Zheng's avatar
Ningxin Zheng committed
22
    :members:
23
24
25
26
27
28
29
30
31
32

..  autoclass:: nni.compression.torch.utils.mask_conflict.CatMaskPadding
    :members:

..  autoclass:: nni.compression.torch.utils.mask_conflict.GroupMaskConflict
    :members:

..  autoclass:: nni.compression.torch.utils.mask_conflict.ChannelMaskConflict
    :members:

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

## Model FLOPs/Parameters Counter

```eval_rst
..  autofunction:: nni.compression.torch.utils.counter.count_flops_params

```