JDAI-CV
diff --git a/‎.clang-format‎
Lines changed: 1 addition & 0 deletions b/‎.clang-format‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎common/daq.fbs‎
Lines changed: 1 addition & 0 deletions b/‎common/daq.fbs‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎dnnlibrary/src/DaqReader.cpp‎
Lines changed: 13 additions & 0 deletions b/‎dnnlibrary/src/DaqReader.cpp‎
Lines changed: 13 additions & 0 deletions
@@ -3,3 +3,4 @@ IndentWidth: 4
 AllowShortBlocksOnASingleLine: false
 AllowShortLoopsOnASingleLine: false
 AllowShortFunctionsOnASingleLine: false
+AlignEscapedNewlines: Left
@@ -190,6 +190,7 @@ table Model {
     initializers:[Tensor];
     inputs:[Input];
     quant_infos:[QuantInfo];
+    outputs:[string];
 }
 
 root_type Model;
@@ -177,6 +177,18 @@ void AddInputs(const DNN::Model &model, ModelBuilder &builder) {
     }
 }
 
+void AddOutputs(const DNN::Model &model, ModelBuilder &builder) {
+    using namespace android::nn::wrapper;
+    if (model.outputs() == nullptr) {
+        return;
+    }
+    for (const auto &output : *model.outputs()) {
+        PNT(output->str());
+        css output_name = output->str();
+        builder.AddOutput(output_name);
+    }
+}
+
 void AddLayers(const DNN::Model &model, ModelBuilder &builder) {
     for (const auto layer : *model.layers()) {
         switch (layer->type()) {
@@ -359,5 +371,6 @@ void ReadDaqImpl(const uint8_t *buf, ModelBuilder &builder) {
     AddInitializersFromBuffer(*model, builder);
     AddInputs(*model, builder);
     AddLayers(*model, builder);
+    AddOutputs(*model, builder);
 }
 }  // namespace dnn
Original file line number	Diff line number	Diff line change
`@@ -190,6 +190,7 @@ table Model {`
`190`	`190`	`initializers:[Tensor];`
`191`	`191`	`inputs:[Input];`
`192`	`192`	`quant_infos:[QuantInfo];`
	`193`	`+ outputs:[string];`
`193`	`194`	`}`
`194`	`195`
`195`	`196`	`root_type Model;`