onnxruntime
84de14a8 - Register OpSet13 CUDA Kernels for BERT/UniLMv2 (#4856)

Commit
5 years ago
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>
Author
Parents
Loading