Add support for boolean outputs in model runner

Comparison operators produce boolean outputs, which need to be written into client data

Allow subgraph traverser to use main block to look for tensors when serialization handler is missing

Signed-off-by: Jiacheng Liang <jiacheng.liang@arm.com>
Change-Id: I6f9af470185541fa6466b3f7786c48f1555fa6f6

1 files changed, 14 insertions, 4 deletions

diff --git a/reference_model/src/subgraph_traverser.cc b/reference_model/src/subgraph_traverser.cc
index 4508291..c02581f 100644
--- a/reference_model/src/subgraph_traverser.cc
+++ b/reference_model/src/subgraph_traverser.cc
@@ -123,16 +123,26 @@ int SubgraphTraverser::initializeGraph()
 
     std::vector<TosaSerializationTensor*> ser_tensor_vec;
     // Get all the serialized tensors from TosaSerializationHandler.
-    for (auto region : tsh->GetRegions())
+    if (tsh)
     {
-        for (auto block : region->GetBlocks())
+        for (auto region : tsh->GetRegions())
         {
-            for (auto ser_tensor : block->GetTensors())
+            for (auto block : region->GetBlocks())
             {
-                ser_tensor_vec.push_back(ser_tensor);
+                for (auto ser_tensor : block->GetTensors())
+                {
+                    ser_tensor_vec.push_back(ser_tensor);
+                }
             }
         }
     }
+    else
+    {
+        for (auto ser_tensor : block->GetTensors())
+        {
+            ser_tensor_vec.push_back(ser_tensor);
+        }
+    }
 
     std::vector<GraphNode*> non_const_node_vec;
     for (auto op : block->GetOperators())