transformers/utils
Julien Plu c8d3fa0dfd
Check TF ops for ONNX compliance (#10025)
* Add check-ops script

* Finish to implement check_tf_ops and start the test

* Make the test mandatory only for BERT

* Update tf_ops folder

* Remove useless classes

* Add the ONNX test for GPT2 and BART

* Add a onnxruntime slow test + better opset flexibility

* Fix test + apply style

* fix tests

* Switch min opset from 12 to 10

* Update src/transformers/file_utils.py

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>

* Fix GPT2

* Remove extra shape_list usage

* Fix GPT2

* Address Morgan's comments

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
2021-02-15 07:55:10 -05:00
..
tf_ops Check TF ops for ONNX compliance (#10025) 2021-02-15 07:55:10 -05:00
check_copies.py Check copies match full class/function names (#10030) 2021-02-08 04:58:25 -05:00
check_dummies.py Transformers fast import part 2 (#9446) 2021-01-07 09:36:14 -05:00
check_repo.py Deprecate Wav2Vec2ForMaskedLM and add Wav2Vec2ForCTC (#10089) 2021-02-09 03:49:02 -05:00
check_table.py Check table as independent script (#8976) 2020-12-07 19:55:12 -05:00
check_tf_ops.py Check TF ops for ONNX compliance (#10025) 2021-02-15 07:55:10 -05:00
download_glue_data.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
get_modified_files.py a more reliable version of branching point discovery (#9449) 2021-01-07 04:47:50 -05:00
link_tester.py Copyright (#8970) 2020-12-07 18:36:34 -05:00
style_doc.py Fix 9918 (#9932) 2021-02-02 05:22:20 -05:00