onnxruntime/onnxruntime/test/python/transformers
Tianlei Wu ef36488df0
Add BeamSearch operator for GPT-2 decoding (#9680)
* Add BeamSearch operator and CPU implementation
* Add ONNX conversion script
2021-12-16 16:08:05 -08:00
..
test_data [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
bert_model_generator.py Attention kernel update to handle different Q,K,V hidden sizes (#8039) 2021-07-19 12:21:33 -07:00
conftest.py
gpt2_model_generator.py update gpt2 attention fusion for past pattern (#8375) 2021-07-14 12:04:53 -07:00
model_loader.py Fuse attention for BERT without num_heads, hidden_size (#9108) 2021-09-20 12:14:33 -07:00
parity_utilities.py Add option to disable FastGelu half2 cuda kernel (#8819) 2021-08-25 08:37:41 -07:00
test_attention_fusion.py Fuse attention for BERT without num_heads, hidden_size (#9108) 2021-09-20 12:14:33 -07:00
test_beam_search.py Add BeamSearch operator for GPT-2 decoding (#9680) 2021-12-16 16:08:05 -08:00
test_gelu_fusions.py refine API of transformer optimizer (#8633) 2021-08-09 10:55:49 -07:00
test_gpt2.py refine API of transformer optimizer (#8633) 2021-08-09 10:55:49 -07:00
test_optimizer.py embed layer fusion gpt2 (#9336) 2021-10-28 11:06:26 -07:00
test_parity_decoder_attention.py Decoder Attention CUDA Op (#9792) 2021-11-19 19:25:36 -08:00
test_parity_gelu.py Add a formula for layernorm parity test (#8888) 2021-08-30 21:22:45 -07:00
test_parity_huggingface_gpt_attention.py Add parity test for LayerNormalization (#8622) 2021-08-05 10:11:19 -07:00
test_parity_layernorm.py Add a formula for layernorm parity test (#8888) 2021-08-30 21:22:45 -07:00
test_profiler.py refine API of transformer optimizer (#8633) 2021-08-09 10:55:49 -07:00
test_shape_infer_helper.py refine API of transformer optimizer (#8633) 2021-08-09 10:55:49 -07:00