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())