Use all outputs if no filter is given

microsoft · Apr 15, 2019 · 7554190 · 7554190
1 parent 69024e4
commit 7554190
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/onnxruntime/hosting/executor.cc b/onnxruntime/hosting/executor.cc
@@ -100,6 +100,13 @@ protobufutil::Status Executor::Predict(const std::string& model_name,
   for (const auto& name : request.output_filter()) {
     output_names.push_back(name);
   }
+
+  // If output_filter is empty then use all outputs
+  if (request.output_filter().empty()) {
+    for (const auto& name : request.output_filter()) {
+      output_names.push_back(name);
+    }
+  }
   std::vector<onnxruntime::MLValue> outputs(output_names.size());
 
   // Run