onnxruntime/onnxruntime/contrib_ops/cuda
2019-11-01 13:49:55 -07:00
..
activation Gelu fusion - kernel and transformer (#1746) 2019-10-03 19:34:46 +08:00
bert Fix mask in EmbedLayerNormalization (#2300) 2019-11-01 13:49:55 -07: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 Add layernorm operator (#1967) 2019-10-03 11:32:13 -07: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