From 4e9a977ae5c95e2a0d323951a8cffcade9b0cbba Mon Sep 17 00:00:00 2001 From: Tai Ly Date: Thu, 16 Mar 2023 22:24:05 +0000 Subject: [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 Change-Id: Icf935561f9f5db38767ff76410bcd36896119395 --- reference_model/src/subgraph_traverser.cc | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'reference_model/src/subgraph_traverser.cc') 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 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); + } } } -- cgit v1.2.1