onnxruntime/onnxruntime/contrib_ops/cuda
Du Li 2659f205cc
Complex multiplication and conjugate contrib ops (#3384)
* adding ComplexMulConj

* Adding fp16 support.

* adding a util func
2020-04-23 17:21:48 -07:00
..
activation Replacing CudaAsyncBuffer with TArray to improve perf (#3303) 2020-03-24 12:13:27 -07:00
bert Update Attention operator for GPT2 (#3474) 2020-04-17 16:20:40 -07:00
math Complex multiplication and conjugate contrib ops (#3384) 2020-04-23 17:21:48 -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
inverse.cc Implement Inverse(12) for CPU and CUDA (#3485) 2020-04-18 17:10:21 -07:00
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
quantize_ops.cc Make De/QuantizeLinear support half (#3531) 2020-04-17 12:17:48 -07:00