The following table highlights the features and frameworks supported by the Vitis AI Optimizer:
Framework | Versions | Features | ||
---|---|---|---|---|
Iterative | One-step | OFA | ||
PyTorch | Supports 1.4 - 1.13, 2.0 | Yes | Yes | Yes |
TensorFlow 1.15 | Supports 1.15 | Yes | No | No |
TensorFlow 2.x | Supports 2.4 - 2.12 | Yes | No | No |
Note:
Vitis AI
2.5 and later releases do not support Caffe and Darknet. Use an earlier release for
these frameworks.