onnxruntime/tools
Xinya Zhang eb827bd3e5
[ROCm] NGramRepeatBlock, LongformerAttention and DecoderAttention Ops (#11971)
* [ROCm] enable NGramRepeatBlock Op

* [ROCm] Enable testing ROCm in NGramRepeatBlockTest.NGramSize_3

Also link onnxruntime_test_all with amdhip64 when USE_ROCM=1

* [ROCm] add LongformerAttention Op

* [ROCm] Enable LongformerAttentionTest

* [ROCm] Add DecoderAttention Op

* Enable DecoderAttention Test for ROCm.

* [ROCM] Updates according to reviews
2022-08-11 19:32:08 -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 [ROCm] NGramRepeatBlock, LongformerAttention and DecoderAttention Ops (#11971) 2022-08-11 19:32:08 -07:00
doc Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
nuget Remove preview keyword from DirectML pacakge (#12368) 2022-07-28 14:18:58 -07:00
perf_view fix json format (#11046) 2022-03-30 16:15:33 -07:00
python Update script to find optimizers that potentially need supported opset updates (#12330) 2022-08-04 07:37:27 +10:00