The following tables shows the features that are supported by the Vitis AI pruner for different frameworks:
| Framework | Versions | Features | ||
|---|---|---|---|---|
| Iterative | One-step | OFA | ||
| PyTorch | Supports 1.4 - 1.9.1 | Yes | Yes | Yes |
| TensorFlow 1.15 | Supports 1.15 | Yes | No | No |
| TensorFlow 2.x | Supports 2.3 - 2.6.2 | Yes | No | No |
| Caffe | - | Yes | No | No |
| Darknet | - | Yes | No | No |