|
3 | 3 |
|
4 | 4 | #pragma once
|
5 | 5 |
|
| 6 | +#include <string> |
6 | 7 | #include <memory>
|
| 8 | +#include <unordered_map> |
7 | 9 | #include <unordered_set>
|
8 | 10 | #include <vector>
|
9 | 11 |
|
10 | 12 | #include "core/common/inlined_containers.h"
|
11 | 13 | #include "core/framework/session_options.h"
|
| 14 | +#include "core/framework/tensor.h" |
12 | 15 | #include "core/optimizer/graph_transformer.h"
|
13 | 16 | #include "core/platform/threadpool.h"
|
14 | 17 |
|
@@ -51,7 +54,8 @@ InlinedVector<std::unique_ptr<GraphTransformer>> GenerateTransformers(
|
51 | 54 | const SessionOptions& session_options,
|
52 | 55 | const IExecutionProvider& execution_provider /*required by constant folding*/,
|
53 | 56 | const InlinedHashSet<std::string>& rules_and_transformers_to_disable = {},
|
54 |
| - concurrency::ThreadPool* intra_op_thread_pool = nullptr); |
| 57 | + concurrency::ThreadPool* intra_op_thread_pool = nullptr, |
| 58 | + std::unordered_map<std::string, std::unique_ptr<Tensor>>* p_buffered_tensors = nullptr); |
55 | 59 |
|
56 | 60 | #endif // !defined(ORT_MINIMAL_BUILD)
|
57 | 61 |
|
@@ -81,7 +85,8 @@ InlinedVector<std::unique_ptr<GraphTransformer>> GenerateTransformersForMinimalB
|
81 | 85 | const SatApplyContextVariant& apply_context,
|
82 | 86 | const IExecutionProvider& cpu_execution_provider,
|
83 | 87 | const InlinedHashSet<std::string>& rules_and_transformers_to_disable = {},
|
84 |
| - concurrency::ThreadPool* intra_op_thread_pool = nullptr); |
| 88 | + concurrency::ThreadPool* intra_op_thread_pool = nullptr, |
| 89 | + std::unordered_map<std::string, std::unique_ptr<Tensor>>* p_buffered_tensors = nullptr); |
85 | 90 |
|
86 | 91 | #endif // !defined(ORT_MINIMAL_BUILD) || defined(ORT_EXTENDED_MINIMAL_BUILD)
|
87 | 92 |
|
|
0 commit comments