onnxruntime/tools
Ye Wang 89ac61f4d4
support gpt2 model with greedy search (#12068)
* greedy search gpt2 cpu checkin

* add cuda support

* add test

* provider

* update

* fix some bugs

* refactor impl class

* refactor test

* remove unused func

* refactor parameters class

* simplify padding

* fix lint warnings

* python format

* Revert "python format"

This reverts commit f25fe1017fa33d960b2418ebbb5dba6a4bd043cf.

* python format

* fix pipelines

* fix pipeline

* move bufferallocater to generate_impl_base

* review comments(alignment, filename/namespace change)

* rebase2

* python reformat

* reformat

* fix rocm build

* review comment

* review comments

* review comments

* fix a bug

* rebase test files

* python format

* format import order

* review comments

* fix build
2022-07-22 15:45:16 -07:00
..
android_custom_build Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
ci_build support gpt2 model with greedy search (#12068) 2022-07-22 15:45:16 -07:00
doc Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
natvis Refactor transformers and other code to reduce memory allocation calls (#10523) 2022-02-24 16:17:14 -08:00
nuget DML EP Update to DML 1.9 (#12090) 2022-07-05 16:30:54 -07:00
perf_view fix json format (#11046) 2022-03-30 16:15:33 -07:00
python Set black's target version (#11370) 2022-04-27 14:52:19 -07:00