From eb52cc18b342d6329322f84b671eab4450e663fd Mon Sep 17 00:00:00 2001 From: Jiacheng Liang Date: Wed, 17 May 2023 16:49:44 +0100 Subject: 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 Change-Id: I6f9af470185541fa6466b3f7786c48f1555fa6f6 --- reference_model/src/subgraph_traverser.cc | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 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 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 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 non_const_node_vec; for (auto op : block->GetOperators()) -- cgit v1.2.1