transformers/tests/onnx
Dean Wyatte a26d71d6ae
Export TensorFlow models to ONNX with dynamic input shapes (#19255)
* validate onnx models with a different input geometry than saved with

* only test working features for now

* simpler test skipping

* rm TODO

* expose batch_size/seq_length on vit

* skip certain name, feature, framework parameterizations known to fail validation

* Trigger CI

* Trigger CI
2022-10-07 10:53:03 -04:00
..
__init__.py
test_features.py Determine framework automatically before ONNX export (#18615) 2022-08-25 16:31:34 +02:00
test_onnx_v2.py Export TensorFlow models to ONNX with dynamic input shapes (#19255) 2022-10-07 10:53:03 -04:00
test_onnx.py