saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-15 01:23:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/python/tools/microbench
History
zhangyaobit 9cbcc93e03
Add micro-benchmarks for Attention and SkipLayerNormalization ops. (#10798) 
* Add micro-benchmarks for Attention and SkipLayerNormalization ops.

* Add choices for argument provider and precision.

* Automatically select CUDA or ROCM execution provider.
2022-03-09 18:18:51 -08:00
..
models Add micro-benchmarks for Attention and SkipLayerNormalization ops. (#10798) 2022-03-09 18:18:51 -08:00
attention.py Add micro-benchmarks for Attention and SkipLayerNormalization ops. (#10798) 2022-03-09 18:18:51 -08:00
benchmark.py Add micro-benchmarks for Attention and SkipLayerNormalization ops. (#10798) 2022-03-09 18:18:51 -08:00
fast_gelu.py Refactor the common code per operator into an abstract base class. (#10785) 2022-03-07 13:15:49 -08:00
matmul.py Refactor the common code per operator into an abstract base class. (#10785) 2022-03-07 13:15:49 -08:00
skip_layer_norm.py Add micro-benchmarks for Attention and SkipLayerNormalization ops. (#10798) 2022-03-09 18:18:51 -08:00