onnxruntime/include/onnxruntime/core/graph
Sherlock eb0f57f0e4
Localized Recompute for Gelu and AttentionDropout (#4402)
* Gelu Activation Recompute Draft

* Prototype for localized recompute

* Introduce localized_recompute rewriter

* Command line args for enabling recompute

* Add logger to Gradient Graph Builder

* use const when possible
2020-08-04 21:48:15 -07:00
..
basic_types.h Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
constants.h Added GetAvailableProviders() to C API (#4247) 2020-06-22 10:10:25 +08:00
function.h Avoid using the default logger in the graph lib and optimizers (#2361) 2019-11-14 13:23:28 -08:00
graph.h support pipeline partition with shared initializer (#4321) 2020-07-14 17:21:40 -07:00
graph_nodes.h Add pipeline online partition logic for pipeline (#3996) 2020-05-26 17:44:09 -07:00
graph_viewer.h Localized Recompute for Gelu and AttentionDropout (#4402) 2020-08-04 21:48:15 -07:00
indexed_sub_graph.h Document the Graph header files and cleanup some issues. (#42) 2018-11-28 08:42:11 -08:00
node_arg.h Enable static memory planning for pipeline. (#4204) 2020-06-12 21:43:50 -07:00
onnx_protobuf.h Fix static analysis warnings found by VC++ (#3530) 2020-04-16 01:46:47 -07:00
schema_registry.h Provide alternative std::mutex implementation on Windows (#3000) 2020-02-11 11:46:08 -08:00