aboutsummaryrefslogtreecommitdiff
path: root/applications/inference_process/include/inference_process.hpp
diff options
context:
space:
mode:
authorKristofer Jonsson <kristofer.jonsson@arm.com>2020-09-10 13:26:41 +0200
committerKristofer Jonsson <kristofer.jonsson@arm.com>2020-09-15 13:27:50 +0200
commit72fa50bcf362643431c39642e5af30781714b2fc (patch)
tree4c8234b1f5a76d898991379fcdd6441eff3d18b0 /applications/inference_process/include/inference_process.hpp
parent98e379c83dd24619752e72e7aefdc15484813652 (diff)
downloadethos-u-core-software-72fa50bcf362643431c39642e5af30781714b2fc.tar.gz
Support inferences with multiple inputs and outputs
Update inference process apis to support inferences with multiple inputs and multiple outputs. Update message process to handle new inference request message with an array of input- and output buffers. Change-Id: Ide0897385a1d829f58edace79140d01d8e3b85a3
Diffstat (limited to 'applications/inference_process/include/inference_process.hpp')
-rw-r--r--applications/inference_process/include/inference_process.hpp13
1 files changed, 7 insertions, 6 deletions
diff --git a/applications/inference_process/include/inference_process.hpp b/applications/inference_process/include/inference_process.hpp
index a5fef2c..53b9331 100644
--- a/applications/inference_process/include/inference_process.hpp
+++ b/applications/inference_process/include/inference_process.hpp
@@ -21,6 +21,7 @@
#include <queue>
#include <stdlib.h>
#include <string>
+#include <vector>
namespace InferenceProcess {
struct DataPtr {
@@ -33,17 +34,17 @@ struct DataPtr {
struct InferenceJob {
std::string name;
DataPtr networkModel;
- DataPtr input;
- DataPtr output;
- DataPtr expectedOutput;
+ std::vector<DataPtr> input;
+ std::vector<DataPtr> output;
+ std::vector<DataPtr> expectedOutput;
size_t numBytesToPrint;
InferenceJob();
InferenceJob(const std::string &name,
const DataPtr &networkModel,
- const DataPtr &input,
- const DataPtr &output,
- const DataPtr &expectedOutput,
+ const std::vector<DataPtr> &input,
+ const std::vector<DataPtr> &output,
+ const std::vector<DataPtr> &expectedOutput,
size_t numBytesToPrint);
};