saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/contrib_ops
History
Yufeng Li 2536553136
use cublasHgemm for Volta GPU (#2074) 
* use cublasHgemm for Volta GPU
2019-10-14 17:29:13 -07:00
..
cpu Move access to intra-op threadpool into OpKernelContext. (#2091) 2019-10-11 10:36:20 -07:00
cuda use cublasHgemm for Volta GPU (#2074) 2019-10-14 17:29:13 -07:00
cpu_contrib_kernels.cc Add layernorm operator (#1967) 2019-10-03 11:32:13 -07:00
cpu_contrib_kernels.h move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00
cuda_contrib_kernels.cc Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -07:00
cuda_contrib_kernels.h move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00