Supported Frameworks and Features - 2.5 English

Vitis AI Optimizer User Guide (UG1333)

Document ID
Release Date
2.5 English

The following tables shows the features that are supported by the Vitis AI pruner for different frameworks:

Table 1. Frameworks and Features Supported by Vitis AI Pruner
Framework Versions Features
Iterative One-step OFA
PyTorch Supports 1.4 - 1.10 Yes Yes Yes
TensorFlow 1.15 Supports 1.15 Yes No No
TensorFlow 2.x Supports 2.3 - 2.8.0 Yes No No
Note: Vitis AI 2.5 and later releases do not support Caffe and Darknet. To use the Vitis AI Pruner on these two frameworks, use an earlier version of Vitis AI.