mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-25 19:48:11 +00:00
* Add Attention op for multi head self attention in BERT * Add test cases * Move op from kOnnxDomain to kMSDomain. Limit test to run by CUDA provider only. * fix test * Add float16 test * fix cpu build error * handle cuda error * get last cuda error when failed |
||
|---|---|---|
| .. | ||
| automl/featurizers/src/FeaturizerPrep | ||
| codegen | ||
| common | ||
| framework | ||
| graph | ||
| language_interop_ops | ||
| mlas | ||
| optimizer | ||
| platform | ||
| protobuf | ||
| providers | ||
| session | ||
| util | ||