[reference model] support multiple regions

This allows IF/WHILE serialization to use regions
instead of blocks to serialize nested regions.
For backward compatibility, both region and block
serialization are supported for IF/WHILE ops.

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Icf935561f9f5db38767ff76410bcd36896119395

1 files changed, 6 insertions, 3 deletions

diff --git a/reference_model/src/subgraph_traverser.cc b/reference_model/src/subgraph_traverser.cc
index 8867ada..e7641ba 100644
--- a/reference_model/src/subgraph_traverser.cc
+++ b/reference_model/src/subgraph_traverser.cc
@@ -123,11 +123,14 @@ int SubgraphTraverser::initializeGraph()
 
     std::vector<TosaSerializationTensor*> ser_tensor_vec;
     // Get all the serialized tensors from TosaSerializationHandler.
-    for (auto block: tsh->GetMainRegion()->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);
+            }
         }
     }