tests/onnx/test_onnx_v2.py · 56a55d3ce4dc6b1e0df1bb1415d2c62bc575fb7b · chenpangpang / transformers

support ONNX export of XDropout in deberta{,_v2} and sew_d (#17502) · 9d7b70bc

Gary Miguel authored Aug 03, 2022

* support ONNX export of XDropout in deberta{,_v2}

* black

* copy to sew_d

* add test

* isort

* use pytest.mark.filterwarnings

* review comments

9d7b70bc

test_onnx_v2.py 16.4 KB

Replace test_onnx_v2.py