graph_transformer_utils.h File Reference

#include <string>
#include <memory>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include "core/common/inlined_containers.h"
#include "core/framework/session_options.h"
#include "core/framework/tensor.h"
#include "core/optimizer/graph_transformer.h"
#include "core/platform/threadpool.h"
#include "core/optimizer/rule_based_graph_transformer.h"
#include "core/optimizer/rewrite_rule.h"
#include "core/optimizer/selectors_actions/selector_action_transformer_apply_contexts.h"

Include dependency graph for graph_transformer_utils.h:

Go to the source code of this file.

Namespaces
	onnxruntime
	onnxruntime::optimizer_utils

Functions
InlinedVector< std::unique_ptr < RewriteRule > >	onnxruntime::optimizer_utils::GenerateRewriteRules (TransformerLevel level, const InlinedHashSet< std::string > &rules_to_disable={})
std::string	onnxruntime::optimizer_utils::GenerateRuleBasedTransformerName (TransformerLevel level)
std::unique_ptr < RuleBasedGraphTransformer >	onnxruntime::optimizer_utils::GenerateRuleBasedGraphTransformer (TransformerLevel level, const InlinedHashSet< std::string > &rules_to_disable, const InlinedHashSet< std::string_view > &compatible_execution_providers)
InlinedVector< std::unique_ptr < GraphTransformer > >	onnxruntime::optimizer_utils::GenerateTransformers (TransformerLevel level, const SessionOptions &session_options, const IExecutionProvider &execution_provider, const InlinedHashSet< std::string > &rules_and_transformers_to_disable={}, concurrency::ThreadPool *intra_op_thread_pool=nullptr, std::unordered_map< std::string, std::unique_ptr< Tensor >> *p_buffered_tensors=nullptr)
InlinedVector< std::unique_ptr < GraphTransformer > >	onnxruntime::optimizer_utils::GenerateTransformersForMinimalBuild (TransformerLevel level, const SessionOptions &session_options, const SatApplyContextVariant &apply_context, const IExecutionProvider &cpu_execution_provider, const InlinedHashSet< std::string > &rules_and_transformers_to_disable={}, concurrency::ThreadPool *intra_op_thread_pool=nullptr, std::unordered_map< std::string, std::unique_ptr< Tensor >> *p_buffered_tensors=nullptr)