Supported Frameworks and Features - 2.0 English

Vitis AI Optimizer User Guide (UG1333)

Document ID
Release Date
2.0 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.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