onnxruntime/onnxruntime/contrib_ops
2019-10-17 23:10:54 -07:00
..
cpu Fix kernel registry bug (#2137) 2019-10-17 23:10:54 -07:00
cuda use cublasHgemm for Volta GPU (#2074) 2019-10-14 17:29:13 -07:00
cpu_contrib_kernels.cc Fix kernel registry bug (#2137) 2019-10-17 23:10:54 -07:00
cpu_contrib_kernels.h Fix kernel registry bug (#2137) 2019-10-17 23:10:54 -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