onnxruntime/onnxruntime/test/python/transformers
Tianlei Wu 29a818caa0
Attention fusion for stable diffusion clip model (#17445)
Add attention fusion for stable diffusion clip model to improve performance of SD or SDXL
2023-09-08 14:17:14 -07:00
..
test_data fix embed layer norm fusion with embedding sum output (#17460) 2023-09-07 22:01:26 -07:00
benchmark_mha.py Flash Attention v2 MHA (#17227) 2023-08-31 13:52:21 -07:00
benchmark_mha.sh Flash Attention v2 MHA (#17227) 2023-08-31 13:52:21 -07:00
bert_model_generator.py
bert_padding.py Flash Attention v2 MHA (#17227) 2023-08-31 13:52:21 -07:00
conftest.py
gpt2_model_generator.py fix embed layer norm fusion with embedding sum output (#17460) 2023-09-07 22:01:26 -07:00
model_loader.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
parity_utilities.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
test_attention_fusion.py Fix weight tensors in transformers optimizer not saved to external data (#17427) 2023-09-06 13:06:19 -07:00
test_embedlayer_fusion.py fix embed layer norm fusion with embedding sum output (#17460) 2023-09-07 22:01:26 -07:00
test_flash_attn.py Flash Attention v2 MHA (#17227) 2023-08-31 13:52:21 -07:00
test_gelu_fusions.py
test_gemmfastgelu_fusion.py [Better Engineering] Fix N802 lint errors in tests (#16788) 2023-07-21 09:17:34 -07:00
test_generation.py Add Whisper export with beam search test cases (#17228) 2023-08-20 00:58:08 -07:00
test_gpt2_benchmark.py
test_gpt2_to_onnx.py
test_optimizer.py [Better Engineering] Fix N802 lint errors in tests (#16788) 2023-07-21 09:17:34 -07:00
test_optimizer_stable_diffusion.py Attention fusion for stable diffusion clip model (#17445) 2023-09-08 14:17:14 -07:00
test_parity_decoder_attention.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
test_parity_gelu.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
test_parity_huggingface_gpt_attention.py Fix weight tensors in transformers optimizer not saved to external data (#17427) 2023-09-06 13:06:19 -07:00
test_parity_layernorm.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
test_parity_neox_attention.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
test_parity_t5_mha.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
test_profiler.py
test_shape_infer_helper.py
test_skip_layer_norm_fusion.py Fix SkipLayerNorm fusion in transformer optimizer (#17320) 2023-08-30 21:12:18 -07:00
test_whisper.py Fix weight tensors in transformers optimizer not saved to external data (#17427) 2023-09-06 13:06:19 -07:00
test_whisper_timestamp_processor.py clean up transformers scripts (#17179) 2023-08-17 23:14:49 -07:00
whisper_model_generator.py Whisper Model Optimization (#15473) 2023-04-18 17:13:54 -07:00