From 4c2697c687ce0d6d46be481daab7e04440ee1032 Mon Sep 17 00:00:00 2001 From: Chi Lo Date: Mon, 3 Feb 2025 16:21:43 -0800 Subject: [PATCH] add back function --- .../onnxruntime/core/optimizer/graph_transformer_utils.h | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/include/onnxruntime/core/optimizer/graph_transformer_utils.h b/include/onnxruntime/core/optimizer/graph_transformer_utils.h index 31b0f22340..8f7acc8a5e 100644 --- a/include/onnxruntime/core/optimizer/graph_transformer_utils.h +++ b/include/onnxruntime/core/optimizer/graph_transformer_utils.h @@ -58,6 +58,12 @@ InlinedVector> GenerateTransformers( concurrency::ThreadPool* intra_op_thread_pool = nullptr, std::unordered_map>* p_buffered_tensors = nullptr); +/** Generates all predefined transformers for EPs */ +InlinedVector> GenerateTransformersForEP( + const SessionOptions& session_options, + const IExecutionProvider& cpu_execution_provider, /*required by constant folding*/ + const logging::Logger& logger); + #endif // !defined(ORT_MINIMAL_BUILD) #if !defined(ORT_MINIMAL_BUILD) || defined(ORT_EXTENDED_MINIMAL_BUILD)