transformers/tests/onnx
NielsRogge 7e7f743481
Add SegFormer ONNX support (#18006)
* Add ONNX support

* Make height and width dynamic axes

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-08-31 20:58:44 +02:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_features.py Determine framework automatically before ONNX export (#18615) 2022-08-25 16:31:34 +02:00
test_onnx_v2.py Add SegFormer ONNX support (#18006) 2022-08-31 20:58:44 +02:00
test_onnx.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00