saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-06 04:28:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/contrib_ops/cuda/bert
History
Tianlei Wu 54bbbb78ae
Change mask_index input of Attention op to be optional (#3459) 
Change Mask Index to optional
2020-04-12 22:55:37 -07:00
..
attention.cc Change mask_index input of Attention op to be optional (#3459) 2020-04-12 22:55:37 -07:00
attention.h implement CPU contrib OP Attention (#2333) 2019-11-07 17:14:59 -08:00
attention_impl.cu Change mask_index input of Attention op to be optional (#3459) 2020-04-12 22:55:37 -07:00
attention_impl.h Change mask_index input of Attention op to be optional (#3459) 2020-04-12 22:55:37 -07:00
embed_layer_norm.cc Change mask input of EmbedLayerNormalization op to be optional (#2495) 2019-12-01 08:36:06 -08:00
embed_layer_norm.h Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -07:00
embed_layer_norm_impl.cu Add support for loading TensorProtos with external data from optimizer Initializer (#3045) 2020-02-28 13:19:16 -08:00
embed_layer_norm_impl.h Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -07:00
fast_gelu.cc Add CPU implementation for FastGelu operator (#3398) 2020-04-07 00:19:30 -07:00
fast_gelu.h Add FastGelu Cuda Op for Gelu and Add bias fusion (#2293) 2019-11-07 17:05:55 -08:00
fast_gelu_impl.cu Add FastGelu Cuda Op for Gelu and Add bias fusion (#2293) 2019-11-07 17:05:55 -08:00
fast_gelu_impl.h Add FastGelu Cuda Op for Gelu and Add bias fusion (#2293) 2019-11-07 17:05:55 -08:00
layer_norm.cuh Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -07:00
skip_layer_norm.cc CMake changes (#2961) 2020-02-03 19:33:14 -08:00
skip_layer_norm.h Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -07:00
skip_layer_norm_impl.cu Add support for loading TensorProtos with external data from optimizer Initializer (#3045) 2020-02-28 13:19:16 -08:00
skip_layer_norm_impl.h CMake changes (#2961) 2020-02-03 19:33:14 -08:00