The following table highlights the features that are supported by the Vitis AI Optimizer for the supported frameworks:
Framework | Versions | Features | ||
---|---|---|---|---|
Iterative | One-step | OFA | ||
PyTorch | Supports 1.4 - 1.12 | Yes | Yes | Yes |
TensorFlow 1.15 | Supports 1.15 | Yes | No | No |
TensorFlow 2.x | Supports 2.3 - 2.10 | Yes | No | No |
Note:
Vitis AI
2.5 and later releases do not support Caffe and Darknet. To use the Vitis AI Optimizer with these frameworks, use an earlier
version of Vitis AI.