onnxruntime/onnxruntime
Tianlei Wu 7b39f5090c
Add Attention op for multi-head self attention in BERT (#1984)
* 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
2019-10-07 12:22:54 -07:00
..
automl_ops Implement C API entry points for creating and fetching non-standard types to OrtValue (#1714) 2019-09-11 14:52:47 -07:00
contrib_ops Add Attention op for multi-head self attention in BERT (#1984) 2019-10-07 12:22:54 -07:00
core Add Attention op for multi-head self attention in BERT (#1984) 2019-10-07 12:22:54 -07:00
gsl Downgrade compiler to CentOS 4.8.5 (#1985) 2019-10-03 15:40:46 -07:00
python python session.run() fallback to CPU/CUDA provider for EP failures. (#1960) 2019-10-02 02:38:03 -07:00
server add support for empty version and score route (#1995) 2019-10-04 22:53:11 -07:00
test Add Attention op for multi-head self attention in BERT (#1984) 2019-10-07 12:22:54 -07:00
__init__.py logging related fixes. (#1930) 2019-09-25 23:48:23 -07:00
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings