Supported Frameworks and Features

Supported Frameworks and Features - 2.5 English

Vitis AI Optimizer User Guide (UG1333)

Document ID

UG1333

Release Date

2022-06-15

Version

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
Framework	Versions	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.