onnxruntime/onnxruntime/contrib_ops
Dmitri Smirnov 25b3c51661
Introduce PrimitiveType into a Type System along with an integer constant (#2307)
Improve perf by avoiding GetType<T>() calls. Introduce MLTypeCallDispatcher to switch on Input Type. Add Tensor IsType<T>() fast method.
2019-11-08 17:47:06 -08:00
..
cpu Introduce PrimitiveType into a Type System along with an integer constant (#2307) 2019-11-08 17:47:06 -08:00
cuda implement CPU contrib OP Attention (#2333) 2019-11-07 17:14:59 -08:00
cpu_contrib_kernels.cc implement CPU contrib OP Attention (#2333) 2019-11-07 17:14:59 -08:00
cpu_contrib_kernels.h Fix kernel registry bug (#2137) 2019-10-17 23:10:54 -07:00
cuda_contrib_kernels.cc Add FastGelu Cuda Op for Gelu and Add bias fusion (#2293) 2019-11-07 17:05:55 -08:00
cuda_contrib_kernels.h move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00