onnxruntime/onnxruntime/core/graph
Tianlei Wu b2c1937523
Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012)
* Add Embed Layer Normalization and Skip Layer Normalization ops for bert optimization.

* add float16 test for skiplayernorm

* Add test for EmbedLayerNormalization op

* fix cpu build error

* fix build warning

* update HasCudaEnvironment function

* handle cuda error
2019-10-07 17:29:43 -07:00
..
automl_ops Integrate featurizers (#1573) 2019-08-15 13:59:59 -07:00
contrib_ops Add EmbedLayerNormalization and SkipLayerNormalization ops for bert optimization (#2012) 2019-10-07 17:29:43 -07:00
function.cc Replace GSL with GSL-LITE submodule and fix up refs (#1920) 2019-10-01 12:43:29 -07:00
function.h
function_container.h
function_impl.h Fix two warnings in graph lib 2019-04-03 19:41:11 -07:00
function_inliner.h remove graph editor since it's not designed as expected to restrict graph access for rewrite rule. (#119) 2018-12-06 14:10:51 -08:00
graph.cc Downgrade compiler to CentOS 4.8.5 (#1985) 2019-10-03 15:40:46 -07:00
graph_utils.cc Filter out info from non-const initializers during shape inferencing (#1806) 2019-09-26 13:44:33 +10:00
graph_utils.h Filter out info from non-const initializers during shape inferencing (#1806) 2019-09-26 13:44:33 +10:00
graph_viewer.cc Implementation of Nuphar execution provider (#881) 2019-09-01 23:01:47 -07:00
model.cc Replace GSL with GSL-LITE submodule and fix up refs (#1920) 2019-10-01 12:43:29 -07:00
model.h Replace GSL with GSL-LITE submodule and fix up refs (#1920) 2019-10-01 12:43:29 -07:00
op.cc Bump onnx to latest (#1756) 2019-09-12 11:46:49 -07:00
op.h Bump onnx to latest (#1756) 2019-09-12 11:46:49 -07:00
record.h Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
schema_registry.cc Make MemPatternPlanner on/off switchable in model weight loading (#989) 2019-05-16 14:39:09 -07:00