onnxruntime/onnxruntime/contrib_ops/cuda
Tianlei Wu 4bdb5cc8e2
Add CPU implementation for FastGelu operator (#3398)
* Add CPU implementation for FastGelu operator
* Update optimization script  to fuse Gelu or FastGelu according to Elf or Tanh is used in graph.
* Merge BiasGelu and FastGelu into one class
* Enable FastGelu Fusion optimizer for CPU Execution Provider.
2020-04-07 00:19:30 -07:00
..
activation Replacing CudaAsyncBuffer with TArray to improve perf (#3303) 2020-03-24 12:13:27 -07:00
bert Add CPU implementation for FastGelu operator (#3398) 2020-04-07 00:19:30 -07:00
math merge training kernels to master (#2999) 2020-02-13 14:52:35 -08:00
tensor move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00
conv_transpose_with_dynamic_pads.cc
conv_transpose_with_dynamic_pads.h
layer_norm.cc Revert an op version change (#3026) 2020-02-18 09:43:18 -08:00
layer_norm.h Add layernorm operator (#1967) 2019-10-03 11:32:13 -07:00
layer_norm_impl.cu use cublasHgemm for Volta GPU (#2074) 2019-10-14 17:29:13 -07:00
layer_norm_impl.h Add layernorm operator (#1967) 2019-10-03 11:32:13 -07:00