saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-28 20:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/contrib_ops
History
Tianlei Wu b2c1937523
Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 
* Add Embed Layer Normalization and Skip Layer Normalization ops for bert optimization.

* add float16 test for skiplayernorm

* Add test for EmbedLayerNormalization op

* fix cpu build error

* fix build warning

* update HasCudaEnvironment function

* handle cuda error
2019-10-07 17:29:43 -07:00
..
cpu ThreadPool fix for roialign and CropAndResize (#2020) 2019-10-06 22:43:59 -07:00
cuda Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -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