[Serving] Fix function definition issue in processor. (DeepRec-AI#563)

aromazyl · Nov 18, 2022 · 8670c92 · 8670c92
1 parent f4f2d3d
commit 8670c92
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/serving/processor/serving/util.cc b/serving/processor/serving/util.cc
@@ -63,7 +63,7 @@ Status RunOnce(
     const std::vector<string>& target_node_names,
     std::vector<Tensor>* outputs, RunMetadata* run_metadata,
     Session* session,
-    thread::ThreadPoolOptions thread_opt = thread::ThreadPoolOptions()) {
+    thread::ThreadPoolOptions thread_opt) {
   CallableOptions callable_options;
   std::vector<Tensor> feed_tensors;
   *callable_options.mutable_run_options() = run_options;

diff --git a/serving/processor/serving/util.h b/serving/processor/serving/util.h
@@ -12,9 +12,9 @@
 #include "tensorflow/cc/saved_model/constants.h"
 #include "tensorflow/core/platform/protobuf_internal.h"
 #include "tensorflow/core/lib/io/path.h"
+#include "tensorflow/core/lib/core/threadpool_options.h"
 #include "tensorflow/core/util/tensor_bundle/naming.h"
 
-
 namespace tensorflow {
 namespace processor {
 namespace util {
@@ -31,7 +31,8 @@ Status RunOnce(const RunOptions& run_options,
                const std::vector<string>& output_tensor_names,
                const std::vector<string>& target_node_names,
                std::vector<Tensor>* outputs, RunMetadata* run_metadata,
-               Session* session, thread::ThreadPoolOptions thread_opt);
+               Session* session,
+               thread::ThreadPoolOptions thread_opt = thread::ThreadPoolOptions());
 Status RunRestoreCheckpoint(
     bool restore_incr_checkpoint,
     const RunOptions& run_options,