saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-16 21:00:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/include
History
Vincent Wang 84de14a833
Register OpSet13 CUDA Kernels for BERT/UniLMv2 (#4856) 
* opset13 cuda kernels for BERT.

* add opset13 SoftmaxCrossEntropyLoss.

* opset13 size.

* fix argmax/min for ut.

* fix ut failure for argmax/min.

* OrtMemTypeCPUInput

Co-authored-by: Vincent Wang <weicwang@microsoft.com>
2020-09-05 08:09:52 +08:00
..
onnxruntime/core Register OpSet13 CUDA Kernels for BERT/UniLMv2 (#4856) 2020-09-05 08:09:52 +08:00